frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix for failure in D68425364 (#145304)

Browse Source 
Summary: Back out change from #145166 which causes an internal model to fail.

Differential Revision: D68459095

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145304
Approved by: https://github.com/izaitsevfb
This commit is contained in:
Aaron Orenstein
2025-01-22 23:33:02 +00:00
committed by PyTorch MergeBot
parent e6a84be3d3
commit 35c8c31f11
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_fx.py
View File

@ -2768,7 +2768,7 @@ class TestFX(JitTestCase):
return self.other(x)
traced = symbolic_trace(ReturnTypeModule())
self.assertIn("-> list[str]", traced._code)
self.assertIn("-> typing_List[str]", traced._code)
scripted = torch.jit.script(traced)
self.assertIn("-> List[str]", scripted.code)
@ -3567,8 +3567,8 @@ class TestFX(JitTestCase):
traced(x, y)
FileCheck().check("tuple[()]") \
.check("tuple[str,tuple[()]]") \
FileCheck().check("typing_Tuple[()]") \
.check("typing_Tuple[str,typing_Tuple[()]]") \
.run(traced.code)
scripted = torch.jit.script(traced)