Fabian Keller 394c339691 improve typings in unflatten (#141817) ... A first follow-up to https://github.com/pytorch/pytorch/pull/115074 / https://github.com/pytorch/pytorch/pull/141240 following the strategy discussed there (https://github.com/pytorch/pytorch/pull/115074#issuecomment-2480992230). This PR improves the type annotations around `unflatten.py` which had been inaccurate due to the previously suppressed type checking on `torch.nn.Module`. CC @Skylion007 @ezyang Pull Request resolved: https://github.com/pytorch/pytorch/pull/141817 Approved by: https://github.com/Skylion007		2024-11-30 22:12:15 +00:00
..
experimental
passes
__init__.py
_draft_export.py	[export] Improve stacktrace filtering (#141285)	2024-11-22 20:55:04 +00:00
_remove_auto_functionalized_pass.py
_remove_effect_tokens_pass.py
_safeguard.py
_swap.py
_trace.py	[BE] Make maybe_aliasing_or_mutating proper tag (#131990)	2024-11-24 00:12:49 +00:00
_tree_utils.py
_unlift.py
custom_obj.py
decomp_utils.py
dynamic_shapes.py
exported_program.py	Revert "Add truediv support in export serializer (#136364)"	2024-11-27 03:24:31 +00:00
graph_signature.py	Revert "Add truediv support in export serializer (#136364)"	2024-11-27 03:24:31 +00:00
unflatten.py	improve typings in unflatten (#141817)	2024-11-30 22:12:15 +00:00