Skip to content

Commit ee0916e

Browse files
committed
Fix typing in unit test
1 parent 268a14d commit ee0916e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

scripts/02_retrieve_metadata.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from ConfigSpace.configuration_space import Configuration
1212
from ConfigSpace.util import deactivate_inactive_hyperparameters
1313

14+
from autosklearn.data.abstract_data_manager import AbstractDataManager
1415
from autosklearn.constants import *
1516
from autosklearn.metrics import CLASSIFICATION_METRICS, REGRESSION_METRICS
1617
from autosklearn.util import pipeline
@@ -188,9 +189,10 @@ def write_output(outputs, configurations, output_dir, configuration_space, metri
188189
fh.write("%s: %s\n" % (key, description[key]))
189190

190191

191-
class DummyDatamanager():
192+
class DummyDatamanager(AbstractDataManager):
192193
def __init__(self, info, feat_type=None):
193-
self.info = info
194+
super().__init__("Test")
195+
self._info = info
194196
self.feat_type = feat_type
195197

196198
def main():

0 commit comments

Comments
 (0)