Update

mikaylagawarecki · mikaylagawarecki · commit 306e44cbd37f · 2024-06-25T19:56:05.000-07:00
[ghstack-poisoned]
diff --git a/test/test_nn.py b/test/test_nn.py
@@ -1800,7 +1800,7 @@ def test_parameterlistdict_setting_attributes(self):
 
     def test_parameterlistdict_pickle(self):
         # warning from torch.load call in _load_from_bytes used in UntypedStorage.__reduce__
-        WEIGHTS_ONLY_WARN = re.escape("You are using `torch.load` with `weights_only=False`")
+        WEIGHTS_ONLY_WARN = "You are using `torch.load` with `weights_only=False`"
         m = nn.ParameterList(map(nn.Parameter, [torch.rand(2), torch.rand(2)]))
         with self.assertWarnsRegex(FutureWarning, WEIGHTS_ONLY_WARN):
             m = pickle.loads(pickle.dumps(m))
