mypy

Author: xadupre

onnx_diagnostic/export/dynamic_shapes.py

@@ -730,6 +730,7 @@ def process_inputs(
     @property
     def true_model_name(self) -> str:
         "Returns class name or module name."
+        assert self.model is not None, "model was None when the class was initialized."
         return (
             self.model.__class__.__name__
             if isinstance(self.model, torch.nn.Module)
@@ -987,6 +988,10 @@ def move_to_kwargs(
         with the corresponding dynamic shapes.
         *kwargs*, *dynamic_shapes* are modified inplace.
         """
+        assert self.signature is not None, (
+            "model was None when the class was initialized, "
+            "cannot move args to kwargs without the signature."
+        )
         sig = self.signature
         arg_dyn, kw_dyn = dynamic_shapes
         for i, p in enumerate(sig.parameters):

onnx_diagnostic/export/shape_helper.py

@@ -103,7 +103,7 @@ def guess_dynamic_shapes_from_inputs(
         ds = guess_dynamic_shapes_from_inputs([inputs1, inputs2], auto="d")
         pprint.pprint(ds)
 
-    This function returns sometihng equivalent to function
+    This function returns something equivalent to function
     :class:`torch.export.dynamic_shapes.AdditionalInputs` but this
     one needs a model.
 
@@ -120,7 +120,7 @@ def guess_dynamic_shapes_from_inputs(
         ds = torch.export.dynamic_shapes.AdditionalInputs()
         ds.add((), data["inputs"])
         ds.add((), data["inputs2"])
-        pprint.pprint(ds.dynamic_shapes(data["model"], (), inputs1))
+        pprint.pprint(ds.dynamic_shapes(data["model"], (), data["inputs"]))
     """
     mi = ModelInputs(None, inputs)
     return mi.guess_dynamic_shapes(auto=auto)