@@ -150,15 +150,15 @@ def test_load_and_unload_model():
150
150
res_json = json .loads (res )
151
151
assert res_json ['status' ] == 'Workers scaled'
152
152
153
- code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ' , model_name )
153
+ code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0' , model_name )
154
154
assert code == 200 , res
155
155
156
156
code , res = make_unload_model_request (model_name )
157
157
assert code == 200 , res
158
158
res_json = json .loads (res )
159
159
assert res_json ['status' ] == "Model \" {}\" unregistered" .format (model_name ), res
160
160
161
- code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ' , model_name )
161
+ code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0' , model_name )
162
162
assert code == 404 , res
163
163
assert res ['message' ] == "Model not found: {}" .format (model_name ), res
164
164
@@ -184,10 +184,10 @@ def test_load_and_unload_two_models():
184
184
res_json = json .loads (res )
185
185
assert res_json ['status' ] == 'Workers scaled'
186
186
187
- code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ' , model_name_0 )
187
+ code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0' , model_name_0 )
188
188
assert code == 200 , res
189
189
190
- code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ' , model_name_1 )
190
+ code , res = make_invocation_request ('0.0, 0.0, 0.0, 0.0, 0.0' , model_name_1 )
191
191
assert code == 200 , res
192
192
193
193
code , res = make_unload_model_request (model_name_0 )
0 commit comments