diff options
Diffstat (limited to 'pyproject.toml')
-rw-r--r-- | pyproject.toml | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/pyproject.toml b/pyproject.toml index 31f0345..2e7cf28 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,12 +14,12 @@ python = "^3.9" loguru = "^0.6.0" omegaconf = "^2.1.0" einops = "^0.3.0" -pytorch-lightning = "^1.5.9" +pytorch-lightning = "^1.6.3" hydra-core = "^1.1.1" attr = "^0.3.1" -torch = "^1.10.1" -torchvision = "^0.11.2" smart-open = "^5.2.1" +torch = "^1.11.0" +torchvision = "^0.12.0" [tool.poetry.dev-dependencies] pytest = "^5.4.2" @@ -35,7 +35,7 @@ flake8-import-order = "^0.18.1" safety = "^1.9.0" mypy = "^0.770" typeguard = "^2.7.1" -wandb = "^0.12.1" +wandb = "^0.12.16" scipy = "^1.6.1" flake8-annotations = "^2.6.2" flake8-docstrings = "^1.6.0" |