mypy

Author: xadupre

CHANGELOGS.rst

@@ -4,6 +4,7 @@ Change Logs
 0.4.4
 +++++
 
+* :pr:`82`: exposes ``register_flattening_functions``, add option ``--subfolder``
 * :pr:`81`: fixes missing ``intermediate_size`` in configuration
 * :pr:`79`: implements task ``object-detection``
 * :pr:`78`: uses *onnx-weekly* instead of *onnx* to avoid conflicts with *onnxscript*

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -93,6 +93,9 @@ def get_untrained_model_with_inputs(
 
     # model kwagrs
     if dynamic_rope is not None:
+        assert (
+            type(config) is not dict
+        ), f"Unable to set dynamic_rope if the configuration is a dictionary\n{config}"
         config.rope_scaling = (
             {"rope_type": "dynamic", "factor": 10.0} if dynamic_rope else None
         )

onnx_diagnostic/torch_models/test_helper.py

@@ -305,7 +305,7 @@ def validate_model(
             print(f"[validate_model] model_options={model_options!r}")
         print(f"[validate_model] get dummy inputs with input_options={input_options}...")
         summary["model_id"] = model_id
-        summary["model_subfolder"] = subfolder
+        summary["model_subfolder"] = subfolder or ""
 
     iop = input_options or {}
     mop = model_options or {}