machine-learning/hm-mlflow/experiments/classify-mnist/pyproject.toml
[tool.poetry]
package-mode = false
[tool.poetry.dependencies]
python = "3.11.x"
mlflow = "2.16.2"
torch = "2.4.1"
torchvision = "0.19.1"
lightning = "2.4.0"
wandb = "0.17.9"
[tool.poetry.group.dev.dependencies]
poethepoet = "0.29.0"
pytest = "8.3.3"
pytest-cov = "5.0.0"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.poe.tasks]
dev = "python src/main.py"
test = "pytest --verbose --verbose"
test-coverage = "pytest --cov=. --cov-report=xml"
mlflow-ui = "mlflow ui"