fix flake

automl · May 7, 2021 · e8dc64d · e8dc64d
1 parent 1a1918c
commit e8dc64d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/autoPyTorch/api/base_task.py b/autoPyTorch/api/base_task.py
@@ -643,7 +643,7 @@ def _do_traditional_prediction(self, time_left: int, func_eval_time_limit_secs:
                         additional_info.pop('pipeline_configuration')
                         run_history.add(config=configuration, cost=cost,
                                         time=runtime, status=status, seed=self.seed,
-                                        additional_info=additional_info, 
+                                        additional_info=additional_info,
                                         starttime=starttime, endtime=starttime + runtime,
                                         origin=origin)
                     else:

diff --git a/test/test_api/test_api.py b/test/test_api/test_api.py
@@ -1,11 +1,12 @@
 import os
 import pickle
 import sys
-import time
 import unittest
 
 from ConfigSpace.configuration_space import Configuration
 
+from api_utils import dummy_traditional_classification
+
 import numpy as np
 
 import pandas as pd
@@ -18,7 +19,7 @@
 from sklearn.base import clone
 from sklearn.ensemble import VotingClassifier, VotingRegressor
 
-from smac.runhistory.runhistory import RunHistory, StatusType
+from smac.runhistory.runhistory import RunHistory
 
 import torch
 
@@ -30,12 +31,9 @@
     HoldoutValTypes,
 )
 from autoPyTorch.datasets.tabular_dataset import TabularDataset
-from autoPyTorch.evaluation.train_evaluator import TrainEvaluator
 from autoPyTorch.optimizer.smbo import AutoMLSMBO
 from autoPyTorch.pipeline.components.training.metrics.metrics import accuracy
 
-from api_utils import dummy_traditional_classification  # noqa (E402: module level import not   at top of file)
-
 
 # Test
 # ========