@@ -132,7 +132,7 @@ def test_list(self, mock_list):
132
132
133
133
@parameterized .expand (
134
134
[
135
- (None , None , False , None , None , None , None , None ),
135
+ (None , None , False , None , None , None , None , None , True ),
136
136
(
137
137
"odsc-llm-fine-tuning" ,
138
138
None ,
@@ -142,8 +142,9 @@ def test_list(self, mock_list):
142
142
["test.json" ],
143
143
None ,
144
144
None ,
145
+ False ,
145
146
),
146
- (None , "test.gguf" , True , None , ["*.json" ], None , None , None ),
147
+ (None , "test.gguf" , True , None , ["*.json" ], None , None , None , False ),
147
148
(
148
149
None ,
149
150
None ,
@@ -153,6 +154,7 @@ def test_list(self, mock_list):
153
154
["test.json" ],
154
155
None ,
155
156
None ,
157
+ False ,
156
158
),
157
159
(
158
160
None ,
@@ -163,6 +165,7 @@ def test_list(self, mock_list):
163
165
None ,
164
166
{"ftag1" : "fvalue1" },
165
167
{"dtag1" : "dvalue1" },
168
+ False ,
166
169
),
167
170
],
168
171
)
@@ -178,6 +181,7 @@ def test_register(
178
181
ignore_patterns ,
179
182
freeform_tags ,
180
183
defined_tags ,
184
+ ignore_model_artifact_check ,
181
185
mock_register ,
182
186
mock_finish ,
183
187
):
@@ -201,6 +205,7 @@ def test_register(
201
205
ignore_patterns = ignore_patterns ,
202
206
freeform_tags = freeform_tags ,
203
207
defined_tags = defined_tags ,
208
+ ignore_model_artifact_check = ignore_model_artifact_check ,
204
209
)
205
210
)
206
211
result = self .model_handler .post ()
@@ -218,6 +223,7 @@ def test_register(
218
223
ignore_patterns = ignore_patterns ,
219
224
freeform_tags = freeform_tags ,
220
225
defined_tags = defined_tags ,
226
+ ignore_model_artifact_check = ignore_model_artifact_check ,
221
227
)
222
228
assert result ["id" ] == "test_id"
223
229
assert result ["inference_container" ] == "odsc-tgi-serving"
0 commit comments