Skip to content

Commit 094588f

Browse files
committed
fix mypy
1 parent c84d3f7 commit 094588f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

onnx_diagnostic/torch_models/hghub/model_inputs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ def get_untrained_model_with_inputs(
113113
print(f"[get_untrained_model_with_inputs] architectures={archs!r}")
114114
print(f"[get_untrained_model_with_inputs] cls={config.__class__.__name__!r}")
115115
if task is None:
116+
assert archs is not None
116117
task = task_from_arch(archs[0], model_id=model_id, subfolder=subfolder)
117118
if verbose:
118119
print(f"[get_untrained_model_with_inputs] task={task!r}")

0 commit comments

Comments
 (0)