fix ut

xadupre · xadupre · commit f3549c61ded9 · 2025-07-04T17:10:06.000+02:00
diff --git a/_unittests/ut_torch_models/test_hghub_model.py b/_unittests/ut_torch_models/test_hghub_model.py
@@ -17,7 +17,7 @@ class TestHuggingFaceHubModel(ExtTestCase):
     @hide_stdout()
     def test_get_untrained_model_with_inputs_tiny_llm(self):
         mid = "arnir0/Tiny-LLM"
-        data = get_untrained_model_with_inputs(mid, verbose=1)
+        data = get_untrained_model_with_inputs(mid, verbose=1, add_second_input=0)
         self.assertEqual(
             set(data),
             {

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ class TestHuggingFaceHubModel(ExtTestCase):`
`17`	`17`	`@hide_stdout()`
`18`	`18`	`def test_get_untrained_model_with_inputs_tiny_llm(self):`
`19`	`19`	`mid = "arnir0/Tiny-LLM"`
`20`		`- data = get_untrained_model_with_inputs(mid, verbose=1)`
	`20`	`+ data = get_untrained_model_with_inputs(mid, verbose=1, add_second_input=0)`
`21`	`21`	`self.assertEqual(`
`22`	`22`	`set(data),`
`23`	`23`	`{`