fix mypy

Author: xadupre

_unittests/ut_torch_models/test_hghub_model.py

@@ -96,7 +96,7 @@ def test_get_untrained_model_with_inputs_text2text_generation(self):
         data = get_untrained_model_with_inputs(mid, verbose=1)
         self.assertIn((data["size"], data["n_weights"]), [(473928, 118482)])
         model, inputs = data["model"], data["inputs"]
-        raise unittest.SkipTest(f"not wroking for {mid!r}")
+        raise unittest.SkipTest(f"not working for {mid!r}")
         model(**inputs)
 
     @hide_stdout()

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -101,7 +101,7 @@ def reduce_model_config(config: Any, task: str) -> Dict[str, Any]:
     return kwargs
 
 
-def check_hasattr(config: Any, *args: Union[str, Tuple[str, ...]]):
+def check_hasattr(config: Any, *args: Union[str, Tuple[Any, ...]]):
     """
     Checks the confiugation has all the attributes in ``args``.
     Raises an exception otherwise.
@@ -185,7 +185,7 @@ def random_input_kwargs(config: Any, task: str) -> Tuple[Dict[str, Any], Callabl
             ),
             encoder_dim=_pick(config, "n_positions", "d_model"),
         )
-        fct = get_inputs_for_text2text_generation
+        fct = get_inputs_for_text2text_generation  # type: ignore
     elif task == "image-classification":
         check_hasattr(config, "image_size", "num_channels")
         if isinstance(config.image_size, int):