| [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" | |
| pyyaml = "^6.0" | |
| torch = "^2.5.1" | |
| torchvision = "^0.20.1" | |
| transformers = "^4.46.3" | |
| datasets = "^3.1.0" | |
| tensorboard = "^2.18.0" | |
| tqdm = "^4.67.0" | |
| types-tqdm = "^4.67.0.20241119" | |
| [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" | |