From 4d8b9ee81f4afcf6160e8bdf9d0c958ec2d9b138 Mon Sep 17 00:00:00 2001 From: Rachit Kumar Date: Thu, 25 Feb 2021 14:35:57 -0500 Subject: [PATCH] Remove test checking that TPOT-NN errors for multiclass input --- tests/nn_tests.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/tests/nn_tests.py b/tests/nn_tests.py index b9d657cd..fcb6dd97 100644 --- a/tests/nn_tests.py +++ b/tests/nn_tests.py @@ -59,17 +59,6 @@ def test_nn_conf_dict(): clf = TPOTClassifier(config_dict=classifier_config_nn) assert clf.config_dict == classifier_config_nn -def test_nn_errors_on_multiclass(): - """Assert that TPOT-NN throws an error when you try to pass training data with > 2 classes. (NN)""" - clf = TPOTClassifier( - random_state=42, - population_size=1, - generations=1, - config_dict=classifier_config_nn, - template='PytorchLRClassifier' - ) - assert_raises(ValueError, clf.fit, multiclass_X, multiclass_y) - def test_pytorch_lr_classifier(): """Assert that the PytorchLRClassifier model works. (NN)""" clf = nn.PytorchLRClassifier(