Skip to content

Commit 4b6331d

Browse files
zhxchen17facebook-github-bot
authored andcommitted
Improve static typing in tracer. (#3973)
Summary: Pull Request resolved: #3973 X-link: pytorch/pytorch#128552 as title. Reviewed By: angelayi Differential Revision: D58485487 fbshipit-source-id: b2ae3f09e9465fb7f05a49d5a82f491dce5c6fe0
1 parent da9b92d commit 4b6331d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

exir/lowered_backend_module.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from executorch.exir.schema import Program
2525

2626
from executorch.exir.tracer import Value
27+
from torch._library.fake_class_registry import FakeScriptObject
2728

2829
from torch._subclasses import FakeTensor
2930
from torch.export.exported_program import (
@@ -430,7 +431,7 @@ def _get_new_signature( # noqa: C901
430431
) -> Tuple[
431432
ExportGraphSignature,
432433
Dict[str, Union[torch.Tensor, torch.nn.Parameter]],
433-
Dict[str, Union[torch.Tensor, torch.ScriptObject]],
434+
Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]],
434435
]:
435436
"""
436437
Args:

0 commit comments

Comments
 (0)