updating torch version, pytype, and tensorboard

mila-iqia · Mar 11, 2024 · cf4bdeb · cf4bdeb
1 parent 1099db2
commit cf4bdeb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -17,15 +17,16 @@
         'pytest==7.1.2',
         'pytest-cov==3.0.0',
         'pytorch_lightning==1.8.3',
-        'pytype==2023.1.17',
+        'pytype==2024.2.27',
         'sphinx==7.2.6',
         'sphinx-autoapi==3.0.0',
         'sphinx-rtd-theme==1.3.0',
         'sphinxcontrib-napoleon==0.7',
         'sphinxcontrib-katex==0.9.9',
+        'tensorboard==2.16.2',
         'tqdm==4.64.0',
-        'torch==1.12.0',
-        'torchvision==0.13.0',
+        'torch==2.2.1',
+        'torchvision==0.17.1',
     ],
     entry_points={
         'console_scripts': [