Merge pull request #463 from MannLabs/fix-torch-conflict-in-tests

fix tests by adapting to changes in pytorch 2.6
MannLabs · Jan 30, 2025 · b0af1aa · b0af1aa
2 parents 817495f + ca0ddae
commit b0af1aa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/alphadia/workflow/manager.py b/alphadia/workflow/manager.py
@@ -820,7 +820,9 @@ def load_classifier_store(self, path: None | str = None):
 
                     if classifier_hash not in self.classifier_store:
                         classifier = deepcopy(self.classifier_base)
-                        classifier.from_state_dict(torch.load(os.path.join(path, file)))
+                        classifier.from_state_dict(
+                            torch.load(os.path.join(path, file), weights_only=False)
+                        )
                         self.classifier_store[classifier_hash].append(classifier)
 
     def get_classifier(self, available_columns: list, version: int = -1):

diff --git a/tests/unit_tests/test_fdr.py b/tests/unit_tests/test_fdr.py
@@ -280,7 +280,9 @@ def test_feed_forward_save():
 
     new_classifier = fdrx.BinaryClassifierLegacy()
     new_classifier.from_state_dict(
-        torch.load(os.path.join(tempfolder, "test_feed_forward_save.pth"))
+        torch.load(
+            os.path.join(tempfolder, "test_feed_forward_save.pth"), weights_only=False
+        )
     )
 
     y_pred = new_classifier.predict(x)  # noqa: F841  # TODO fix this test