Skip to content

Commit 550f22f

Browse files
author
Malav Shastri
committed
Correct local integration tests
1 parent 15b8350 commit 550f22f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/integration/test_multiple_model_endpoint.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -150,15 +150,15 @@ def test_load_and_unload_model():
150150
res_json = json.loads(res)
151151
assert res_json['status'] == 'Workers scaled'
152152

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)
154154
assert code == 200, res
155155

156156
code, res = make_unload_model_request(model_name)
157157
assert code == 200, res
158158
res_json = json.loads(res)
159159
assert res_json['status'] == "Model \"{}\" unregistered".format(model_name), res
160160

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)
162162
assert code == 404, res
163163
assert res['message'] == "Model not found: {}".format(model_name), res
164164

@@ -184,10 +184,10 @@ def test_load_and_unload_two_models():
184184
res_json = json.loads(res)
185185
assert res_json['status'] == 'Workers scaled'
186186

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)
188188
assert code == 200, res
189189

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)
191191
assert code == 200, res
192192

193193
code, res = make_unload_model_request(model_name_0)

0 commit comments

Comments
 (0)