fix image

xadupre · xadupre · commit 93de2773047d · 2025-03-31T15:15:31.000+02:00
diff --git a/onnx_diagnostic/torch_models/hghub/model_inputs.py b/onnx_diagnostic/torch_models/hghub/model_inputs.py
@@ -241,7 +241,7 @@ def random_input_kwargs(config: Any, task: str) -> Tuple[Dict[str, Any], Callabl
             height=config.vision_config.image_size,
             num_channels=config.vision_config.num_channels,
         )
-        fct = get_inputs_for_image_text_to_text
+        fct = get_inputs_for_image_text_to_text  # type: ignore
     else:
         raise NotImplementedError(f"Input generation for task {task!r} not implemented yet.")
 

Original file line number	Diff line number	Diff line change
`@@ -241,7 +241,7 @@ def random_input_kwargs(config: Any, task: str) -> Tuple[Dict[str, Any], Callabl`
`241`	`241`	`height=config.vision_config.image_size,`
`242`	`242`	`num_channels=config.vision_config.num_channels,`
`243`	`243`	`)`
`244`		`- fct = get_inputs_for_image_text_to_text`
	`244`	`+ fct = get_inputs_for_image_text_to_text # type: ignore`
`245`	`245`	`else:`
`246`	`246`	`raise NotImplementedError(f"Input generation for task {task!r} not implemented yet.")`
`247`	`247`