fix tests after rebase

automl · Dec 6, 2021 · 02ac9de · 02ac9de
1 parent 38471f1
commit 02ac9de
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/test/test_utils/test_results_manager.py b/test/test_utils/test_results_manager.py
@@ -352,6 +352,7 @@ def test_metric_results(metric, scores, ensemble_ends_later):
 
 
 def test_search_results_sprint_statistics():
+    BaseTask.__abstractmethods__ = set()
     api = BaseTask()
     for method in ['get_search_results', 'sprint_statistics', 'get_incumbent_results']:
         with pytest.raises(RuntimeError):

diff --git a/test/test_utils/test_results_visualizer.py b/test/test_utils/test_results_visualizer.py
@@ -146,6 +146,7 @@ def test_set_plot_args(params):  # TODO
 
 @pytest.mark.parametrize('metric_name', ('unknown', 'accuracy'))
 def test_raise_error_in_plot_perf_over_time_in_base_task(metric_name):
+    BaseTask.__abstractmethods__ = set()
     api = BaseTask()
 
     if metric_name == 'unknown':
@@ -159,6 +160,7 @@ def test_raise_error_in_plot_perf_over_time_in_base_task(metric_name):
 @pytest.mark.parametrize('metric_name', ('balanced_accuracy', 'accuracy'))
 def test_plot_perf_over_time(metric_name):  # TODO
     dummy_history = [{'Timestamp': datetime(2022, 1, 1), 'train_accuracy': 1, 'test_accuracy': 1}]
+    BaseTask.__abstractmethods__ = set()
     api = BaseTask()
     run_history_data = json.load(open(os.path.join(os.path.dirname(__file__),
                                                    'runhistory.json'),