Skip to content

Commit

Permalink
Revert "Improve static typing in tracer. (#3973)"
Browse files Browse the repository at this point in the history
This reverts commit 4b6331d.

PT pin not updated to this
  • Loading branch information
kirklandsign committed Jul 25, 2024
1 parent 97f7727 commit e9f8d13
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions exir/lowered_backend_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,6 @@
from executorch.exir.schema import Program

from executorch.exir.tracer import Value
from torch._library.fake_class_registry import FakeScriptObject

from torch._subclasses import FakeTensor
from torch.export.exported_program import (
Expand Down Expand Up @@ -431,7 +430,7 @@ def _get_new_signature( # noqa: C901
) -> Tuple[
ExportGraphSignature,
Dict[str, Union[torch.Tensor, torch.nn.Parameter]],
Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]],
Dict[str, Union[torch.Tensor, torch.ScriptObject]],
]:
"""
Args:
Expand Down

0 comments on commit e9f8d13

Please sign in to comment.