fix

Author: xadupre

onnx_diagnostic/helpers/torch_helper.py

@@ -724,7 +724,7 @@ def to_any(value: Any, to_value: Union[torch.dtype, torch.device, str]) -> Any:
     if value.__class__.__name__ in {"DynamicCache", "HybridCache"}:
         make = dict(DynamicCache=make_dynamic_cache, HybridCache=make_hybrid_cache)
         cc = CacheKeyValue(value)
-        return make[value.__class__.__name__](
+        return make[value.__class__.__name__](  # type: ignore[operator]
             list(
                 zip(
                     [t.to(to_value) if t is not None else t for t in cc.key_cache],

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -239,7 +239,7 @@ def get_untrained_model_with_inputs(
                 ) from e
 
     # input kwargs
-    kwargs, fct = random_input_kwargs(config, task)
+    kwargs, fct = random_input_kwargs(config, task)  # type: ignore[arg-type]
     if verbose:
         print(f"[get_untrained_model_with_inputs] use fct={fct}")
         if os.environ.get("PRINT_CONFIG") in (1, "1"):