Skip to content

Commit

Permalink
[easy] Fix type annotation for ExportedProgram.run_decompositions (p…
Browse files Browse the repository at this point in the history
…ytorch#133720)

Fix the tuple type annotation.

Pull Request resolved: pytorch#133720
Approved by: https://github.com/Skylion007
  • Loading branch information
justinchuby authored and pytorchmergebot committed Aug 16, 2024
1 parent 99e789b commit 271ee90
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/export/exported_program.py
Original file line number Diff line number Diff line change
Expand Up @@ -1005,7 +1005,7 @@ def _num_lifted_params_buffers(self):
def run_decompositions(
self,
decomp_table: Optional[Dict[torch._ops.OperatorBase, Callable]] = None,
_preserve_ops: Tuple[torch._ops.OpOverload] = (), # type: ignore[assignment]
_preserve_ops: Tuple[torch._ops.OpOverload, ...] = (),
) -> "ExportedProgram":
"""
Run a set of decompositions on the exported program and returns a new
Expand Down

0 comments on commit 271ee90

Please sign in to comment.