Prepare project for CI: cleanup dependencies, fix test data location, cleanup assert.

Signed-off-by: Maya Anderson <mayaa@il.ibm.com>
This commit is contained in:
Maya Anderson 2022-12-20 15:26:05 +02:00 committed by abigailgold
parent 805a489201
commit 89bdcfc00e
3 changed files with 19 additions and 15 deletions

View file

@ -219,7 +219,7 @@ def test_blackbox_classifier_no_test_y():
except BaseException:
unable_to_predict_test = True
assert (unable_to_predict_test, True)
assert unable_to_predict_test
def test_blackbox_classifier_no_train_y():
(x_train, _), (x_test, y_test) = dataset_utils.get_iris_dataset_np()
@ -241,7 +241,7 @@ def test_blackbox_classifier_no_train_y():
except BaseException:
unable_to_predict_train = True
assert(unable_to_predict_train,True)
assert unable_to_predict_train
def test_blackbox_classifier_probabilities():
(x_train, _), (_, _) = dataset_utils.get_iris_dataset_np()