Merge pull request #141 from OMalenfantThuot/pyup-update-torchvision-…

…0.7.0-to-0.8.0 Update torchvision to 0.8.0
OMalenfantThuot · Oct 27, 2020 · 5ad3470 · 5ad3470
2 parents bcee482 + f8a032a
commit 5ad3470
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/mlcalcdriver/calculators/schnetpack.py b/mlcalcdriver/calculators/schnetpack.py
@@ -53,13 +53,6 @@ def __init__(
             self.model = load_model(
                 os.environ["MODELDIR"] + model_dir, map_location=self.device
             )
-
-        # Bugfix to make older models work with PyTorch 1.6
-        # Hopefully temporary
-        for mod in self.model.modules():
-            if not hasattr(mod, "_non_persistent_buffers_set"):
-                mod._non_persistent_buffers_set = set()
-
         super(SchnetPackCalculator, self).__init__(units=units)
         self._get_representation_type()
 

diff --git a/requirements_dev.txt b/requirements_dev.txt
@@ -1,11 +1,11 @@
-pip==20.2.3
+pip==20.2.4
 wheel==0.35.1
 black==20.8b1
 numpy==1.19.2
-torch==1.6.0
-torchvision==0.7.0
+torch==1.7.0
+torchvision==0.8.0
 twine==3.2.0
-pytest==6.0.2
+pytest==6.1.1
 pytest-cov==2.10.1
 pytest-sugar==0.9.4
 coveralls==2.1.2