Skip to content

Commit

Permalink
Remove traditional pipeline as it is introduced in PR automl#224
Browse files Browse the repository at this point in the history
  • Loading branch information
ravinkohli committed Jun 9, 2021
1 parent 9270b70 commit 81959b2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/test_api/test_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
from autoPyTorch.datasets.tabular_dataset import TabularDataset
from autoPyTorch.evaluation.abstract_evaluator import DummyClassificationPipeline, DummyRegressionPipeline
from autoPyTorch.optimizer.smbo import AutoMLSMBO
from autoPyTorch.pipeline.components.setup.traditional_ml.traditional_learner import get_available_traditional_learners
from autoPyTorch.pipeline.components.setup.traditional_ml.traditional_learner import _traditional_learners
from autoPyTorch.pipeline.components.training.metrics.metrics import accuracy, r2


Expand Down Expand Up @@ -710,7 +710,7 @@ def test_do_traditional_pipeline(fit_dictionary_tabular):
with open(model_path, 'rb') as model_handler:
model = pickle.load(model_handler)
clone(model)
assert model.config == list(_classifiers.keys())[i - 2]
assert model.config == list(_traditional_learners.keys())[i - 2]
at_least_one_model_checked = True
if not at_least_one_model_checked:
pytest.fail("Not even one single traditional pipeline was fitted")
Expand Down

0 comments on commit 81959b2

Please sign in to comment.