Skip to content

Commit dba0d61

Browse files
author
Dan
authored
change: make multi algo integration test assertion less specific (#1147)
1 parent 391733e commit dba0d61

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_tuner_multi_algo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ def _deploy_and_predict(sagemaker_session, tuner, data_set, cpu_instance_type):
205205

206206
assert len(result["predictions"]) == len(data)
207207
for prediction in result["predictions"]:
208-
assert prediction["predicted_label"] is not None
208+
assert prediction is not None
209209

210210

211211
def _create_training_inputs(sagemaker_session):

0 commit comments

Comments
 (0)