From 372b023eb1df6f565549c75d29c3653204d12a87 Mon Sep 17 00:00:00 2001 From: Mikayla Gawarecki Date: Fri, 20 Dec 2024 10:44:42 -0800 Subject: [PATCH] Fix test_serialization_zipfile_actually_jit when weights_only is not default (#143668) Fails in fbcode where weights_only isn't default Pull Request resolved: https://github.com/pytorch/pytorch/pull/143668 Approved by: https://github.com/awgu ghstack dependencies: #143326, #143403 --- test/test_serialization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_serialization.py b/test/test_serialization.py index cfd465e650c3..302788f7c6a0 100644 --- a/test/test_serialization.py +++ b/test/test_serialization.py @@ -1020,7 +1020,7 @@ class TestSerialization(TestCase, SerializationMixin): RuntimeError, re.escape("Cannot use ``weights_only=True`` with TorchScript archives passed to ``torch.load``") ): - torch.load(f) + torch.load(f, weights_only=True) f.seek(0) torch.load(f, weights_only=False)