File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -128,10 +128,9 @@ def test_v1_3(self):
128128 formats = polyglot .identify_pytorch_file_format (self .filename_v1_3 )
129129 self .assertEqual (formats , ["PyTorch v1.3" ])
130130
131- # NOTE(boyan): this test doesn't pass but it should. This needs to be fixed.
132- # def test_legacy_pickle(self):
133- # formats = polyglot.identify_pytorch_file_format(self.filename_legacy_pickle)
134- # self.assertEqual(formats, ["PyTorch v0.1.10"])
131+ def test_legacy_pickle (self ):
132+ formats = polyglot .identify_pytorch_file_format (self .filename_legacy_pickle )
133+ self .assertEqual (formats , ["PyTorch v0.1.10" ])
135134
136135 @unittest .skipIf (_lacks_torch_jit_support , "PyTorch 2.9.1 JIT broken with Python 3.14+" )
137136 def test_torchscript (self ):
You can’t perform that action at this time.
0 commit comments