Skip to content

Commit

Permalink
[export] Fix export arg type declaration (pytorch#109060)
Browse files Browse the repository at this point in the history
Summary: Its a arbitrary length tuple of anything. Tuple[Any] means 1 element.

Test Plan: ci

Differential Revision: D49161625

Pull Request resolved: pytorch#109060
Approved by: https://github.com/angelayi
  • Loading branch information
JacobSzwejbka authored and pytorchmergebot committed Sep 11, 2023
1 parent b0656ac commit 5c8efa6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/export/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -927,7 +927,7 @@ def dynamic_dim(t: torch.Tensor, index: int):

def export(
f: Callable,
args: Tuple[Any],
args: Tuple[Any, ...],
kwargs: Optional[Dict[str, Any]] = None,
*,
constraints: Optional[List[Constraint]] = None,
Expand Down

0 comments on commit 5c8efa6

Please sign in to comment.