Skip to content

Commit 6b2e4aa

Browse files
update handler tests
1 parent 6aaa3ef commit 6b2e4aa

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/unitary/with_extras/aqua/test_model_handler.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def test_list(self, mock_list):
132132

133133
@parameterized.expand(
134134
[
135-
(None, None, False, None, None, None, None, None),
135+
(None, None, False, None, None, None, None, None, True),
136136
(
137137
"odsc-llm-fine-tuning",
138138
None,
@@ -142,8 +142,9 @@ def test_list(self, mock_list):
142142
["test.json"],
143143
None,
144144
None,
145+
False,
145146
),
146-
(None, "test.gguf", True, None, ["*.json"], None, None, None),
147+
(None, "test.gguf", True, None, ["*.json"], None, None, None, False),
147148
(
148149
None,
149150
None,
@@ -153,6 +154,7 @@ def test_list(self, mock_list):
153154
["test.json"],
154155
None,
155156
None,
157+
False,
156158
),
157159
(
158160
None,
@@ -163,6 +165,7 @@ def test_list(self, mock_list):
163165
None,
164166
{"ftag1": "fvalue1"},
165167
{"dtag1": "dvalue1"},
168+
False,
166169
),
167170
],
168171
)
@@ -178,6 +181,7 @@ def test_register(
178181
ignore_patterns,
179182
freeform_tags,
180183
defined_tags,
184+
ignore_model_artifact_check,
181185
mock_register,
182186
mock_finish,
183187
):
@@ -201,6 +205,7 @@ def test_register(
201205
ignore_patterns=ignore_patterns,
202206
freeform_tags=freeform_tags,
203207
defined_tags=defined_tags,
208+
ignore_model_artifact_check=ignore_model_artifact_check,
204209
)
205210
)
206211
result = self.model_handler.post()
@@ -218,6 +223,7 @@ def test_register(
218223
ignore_patterns=ignore_patterns,
219224
freeform_tags=freeform_tags,
220225
defined_tags=defined_tags,
226+
ignore_model_artifact_check=ignore_model_artifact_check,
221227
)
222228
assert result["id"] == "test_id"
223229
assert result["inference_container"] == "odsc-tgi-serving"

0 commit comments

Comments
 (0)