Skip to content

Commit edab5b8

Browse files
committed
fix ut
1 parent 8037d0c commit edab5b8

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

_unittests/ut_torch_models/test_hghub_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def test_get_untrained_model_with_inputs_tiny_llm(self):
3030
"input_kwargs",
3131
"model_kwargs",
3232
"task",
33+
"dump_info",
3334
},
3435
)
3536
model, inputs = data["model"], data["inputs"]

onnx_diagnostic/torch_models/hghub/model_inputs.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,10 @@ def get_untrained_model_with_inputs(
135135
mkwargs[k] = v
136136
if mkwargs:
137137
update_config(config, mkwargs)
138-
diff_config = build_diff_config(config0, config)
138+
try:
139+
diff_config = build_diff_config(config0, config)
140+
except (ValueError, AttributeError, TypeError) as e:
141+
diff_config = f"DIFF CONFIG ERROR {e}"
139142
if verbose:
140143
if diff_config:
141144
print("[get_untrained_model_with_inputs] -- updated config")

0 commit comments

Comments
 (0)