fix

Author: xadupre

onnx_diagnostic/investigate/input_observer.py

@@ -677,6 +677,8 @@ def _post_process_for_kwargs(self, kwargs: dict[str, Any]) -> dict[str, Any]:
             # Nothing to do here.
             return kwargs
         to_be_moved = {k for k in kwargs if k not in self.signature_names}
+        if not to_be_moved:
+            return kwargs
         keywords = {k: v for k, v in kwargs.items() if k in to_be_moved}
         new_kwargs = {k: v for k, v in kwargs.items() if k not in to_be_moved}
         return {**new_kwargs, self.kwargs_name: keywords}

onnx_diagnostic/torch_export_patches/patches/patch_torch.py

@@ -165,13 +165,8 @@ def patched__get_range_constraints(
     assert isinstance(
         combined_args, dict
     ), f"unexpected type {type(combined_args)} for 'combined_args'"
-    input_names = [
-        s.arg.name
-        for s in export_graph_signature.input_specs
-        if s.kind == torch.export.graph_signature.InputKind.USER_INPUT
-    ]
     new_args = {}
-    for k in input_names:
+    for k in kwargs:
         if k in combined_args:
             new_args[k] = combined_args[k]
     for k in combined_args: