Skip to content

Commit 5db7a56

Browse files
committed
Change one torch.load
Signed-off-by: Eric Kerfoot <[email protected]>
1 parent 3e0b517 commit 5db7a56

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/data/meta_tensor/test_meta_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def test_pickling(self):
245245
with tempfile.TemporaryDirectory() as tmp_dir:
246246
fname = os.path.join(tmp_dir, "im.pt")
247247
torch.save(m, fname)
248-
m2 = torch.load(fname, weights_only=True)
248+
m2 = torch.load(fname, weights_only=False)
249249
self.check(m2, m, ids=False)
250250

251251
@skip_if_no_cuda

0 commit comments

Comments
 (0)