mypy

Author: xadupre

onnx_diagnostic/export/api.py

@@ -479,7 +479,7 @@ def method_to_onnx(
         custom loops (see :func:`onnx_diagnostic.export.control_flow_onnx.loop_for_onnx`)
     :param onnx_plugs: the code was modified to replace some parts with onnx translation
     :param inline: inline local functions
-    :param convert_after_n_calls: convets the model after this number of calls.
+    :param convert_after_n_calls: converts the model after this number of calls.
     :param patch_kwargs: patch arguments
     :return: the output of the selected exporter, usually a structure including
         an onnx model

onnx_diagnostic/export/dynamic_shapes.py

@@ -1000,7 +1000,7 @@ def guess_dynamic_shapes(self, auto: Union[bool, str] = False) -> DYNAMIC_SHAPES
                 msg=lambda name=name: f" failing input {name!r}",
             )
         # reordering
-        if kwargs is not None:
+        if kwargs is not None and self.forward_ordered_parameter_names:
             kwargs = {
                 p: kwargs[p] for p in self.forward_ordered_parameter_names if p in kwargs
             }
@@ -1066,11 +1066,11 @@ def move_to_kwargs(
                 f"and kwargs={set(kwargs)}, "
                 f"forward_ordered_parameter_names={self.forward_ordered_parameter_names}"
             )
-        if kwargs is not None:
+        if kwargs is not None and self.forward_ordered_parameter_names:
             kwargs = {
                 p: kwargs[p] for p in self.forward_ordered_parameter_names if p in kwargs
             }
-        if kw_dyn is not None:
+        if kw_dyn is not None and self.forward_ordered_parameter_names:
             kw_dyn = {
                 p: kw_dyn[p] for p in self.forward_ordered_parameter_names if p in kw_dyn
             }