doc

Author: xadupre

_doc/conf.py

@@ -119,6 +119,8 @@ def linkcode_resolve(domain, info):
     ("py:class", "ast.Node"),
     ("py:class", "dtype"),
     ("py:class", "False"),
+    ("py:class", "FakeTensor"),
+    ("py:class", "FakeTensorMode"),
     ("py:class", "True"),
     ("py:class", "Argument"),
     ("py:class", "CacheProcessor"),

onnx_diagnostic/torch_export_patches/patches/patch_torch.py

@@ -25,8 +25,8 @@ def retrieve_stacktrace():
 
 def _catch_produce_guards_and_solve_constraints(
     previous_function: Callable,
-    fake_mode: "FakeTensorMode",  # noqa: F821
-    gm: "torch.fx.GraphModule",  # noqa: F821
+    fake_mode: FakeTensorMode,
+    gm: torch.fx.GraphModule,
     dynamic_shapes: Union[Dict[str, Any], Tuple[Any], List[Any], None],
     equalities_inputs: "EqualityConstraint",  # noqa: F821
     original_signature: inspect.Signature,