lint: black . isort --profile=black . flake8 --ignore E501 . test: python main.py --alpha 1 --lr 1e-3 --max_epochs 100