fix

xadupre · xadupre · commit 02a6e833b7ba · 2025-04-01T13:52:32.000+02:00
diff --git a/_unittests/ut_torch_models/test_hghub_model.py b/_unittests/ut_torch_models/test_hghub_model.py
@@ -38,6 +38,7 @@ def test_get_untrained_model_with_inputs_tiny_llm(self):
                 "n_weights",
                 "input_kwargs",
                 "model_kwargs",
+                "task",
             },
         )
         model, inputs = data["model"], data["inputs"]

Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,7 @@ def test_get_untrained_model_with_inputs_tiny_llm(self):`
`38`	`38`	`"n_weights",`
`39`	`39`	`"input_kwargs",`
`40`	`40`	`"model_kwargs",`
	`41`	`+ "task",`
`41`	`42`	`},`
`42`	`43`	`)`
`43`	`44`	`model, inputs = data["model"], data["inputs"]`