Skip to content

Commit 7f6f8e6

Browse files
Uncomment passing test_legacy_pickle test (#250)
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent f5dc67b commit 7f6f8e6

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

test/test_polyglot.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff 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):

0 commit comments

Comments
 (0)