| [tool.poetry] | |
| name = "mnist-classifier" | |
| version = "0.1.0" | |
| description = "MNIST classifier using PyTorch" | |
| authors = ["Felix"] | |
| license = "MIT" | |
| readme = "README.md" | |
| [tool.poetry.dependencies] | |
| python = "^3.10" | |
| torch = "^2.4.0" | |
| torchvision = "^0.15.0" | |
| pytorch-lightning = "^2.0.0" | |
| wandb = "^0.15.0" | |
| torchmetrics = "^1.0.0" | |
| datasets = "^2.0.0" | |
| huggingface-hub = "^0.16.0" | |
| pyyaml = "^6.0" | |
| [tool.poetry.group.dev.dependencies] | |
| pytest = "^7.0.0" | |
| black = "^23.0.0" | |
| isort = "^5.12.0" | |
| flake8 = "^6.0.0" | |
| mypy = "^1.0.0" | |
| [build-system] | |
| requires = ["poetry-core"] | |
| build-backend = "poetry.core.masonry.api" | |