Change one torch.load

Signed-off-by: Eric Kerfoot <[email protected]>
Project-MONAI · Feb 25, 2025 · 5db7a56 · 5db7a56
1 parent 3e0b517
commit 5db7a56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/data/meta_tensor/test_meta_tensor.py b/tests/data/meta_tensor/test_meta_tensor.py
@@ -245,7 +245,7 @@ def test_pickling(self):
         with tempfile.TemporaryDirectory() as tmp_dir:
             fname = os.path.join(tmp_dir, "im.pt")
             torch.save(m, fname)
-            m2 = torch.load(fname, weights_only=True)
+            m2 = torch.load(fname, weights_only=False)
         self.check(m2, m, ids=False)
 
     @skip_if_no_cuda