fix mypy

Author: xadupre

onnx_diagnostic/helpers/cache_helper.py

@@ -103,7 +103,7 @@ def make_dynamic_cache(
             )
             print(string_type(past_key_values, with_shape=True))
         """
-        cache = transformers.cache_utils.DynamicCache(len(key_value_pairs))
+        cache = transformers.cache_utils.DynamicCache(len(key_value_pairs))  # type: ignore
         for i, (key, value) in enumerate(key_value_pairs):
             cache.update(key, value, i)
         return cache

onnx_diagnostic/torch_models/hghub/hub_api.py

@@ -1,5 +1,5 @@
 import functools
-from typing import Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 import transformers
 from huggingface_hub import HfApi, model_info
 from . import hub_data_cached_configs
@@ -38,7 +38,7 @@ def get_cached_configuration(name: str) -> Optional[transformers.PretrainedConfi
 
 def get_pretrained_config(
     model_id: str, trust_remote_code: bool = True, use_cached: bool = True
-) -> str:
+) -> Any:
     """
     Returns the config for a model_id.
 
@@ -49,6 +49,7 @@ def get_pretrained_config(
         accessing the network, if available, it is returned by
         :func:`get_cached_configuration`, the cached list is mostly for
         unit tests
+    :return: a configuration
     """
     if use_cached:
         conf = get_cached_configuration(model_id)

onnx_diagnostic/torch_models/test_helper.py

@@ -111,13 +111,15 @@ def filter_inputs(
 
 def _make_folder_name(
     model_id: str,
-    exporter: str,
+    exporter: Optional[str],
     optimization: Optional[str] = None,
     dtype: Optional[Union[str, torch.dtype]] = None,
     device: Optional[Union[str, torch.device]] = None,
 ) -> str:
     "Creates a filename unique based on the given options."
-    els = [model_id.replace("/", "_"), exporter]
+    els = [model_id.replace("/", "_")]
+    if exporter:
+        els.append(exporter)
     if optimization:
         els.append(optimization)
     if dtype is not None and dtype: