Skip to content

Commit 2046154

Browse files
[FIX] flake
1 parent a4853c0 commit 2046154

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

autoPyTorch/evaluation/tae.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import logging
55
import math
66
import multiprocessing
7+
import os
78
import time
89
import traceback
910
import typing
@@ -25,6 +26,7 @@
2526
from autoPyTorch.evaluation.utils import empty_queue, extract_learning_curve, read_queue
2627
from autoPyTorch.pipeline.components.training.metrics.base import autoPyTorchMetric
2728
from autoPyTorch.utils.backend import Backend
29+
from autoPyTorch.utils.common import replace_string_bool_to_bool
2830
from autoPyTorch.utils.hyperparameter_search_space_update import HyperparameterSearchSpaceUpdates
2931
from autoPyTorch.utils.logging_ import PicklableClientLogger, get_named_client_logger
3032

@@ -144,7 +146,12 @@ def __init__(
144146
self.exclude = exclude
145147
self.disable_file_output = disable_file_output
146148
self.init_params = init_params
147-
self.pipeline_config = pipeline_config
149+
self.pipeline_config: typing.Dict[str, typing.Union[int, str, float]] = dict()
150+
if pipeline_config is None:
151+
pipeline_config = replace_string_bool_to_bool(json.load(open(
152+
os.path.join(os.path.dirname(__file__), '../configs/default_pipeline_options.json'))))
153+
self.pipeline_config.update(pipeline_config)
154+
148155
self.budget_type = pipeline_config['budget_type'] if pipeline_config is not None else budget_type
149156
self.logger_port = logger_port
150157
if self.logger_port is None:

0 commit comments

Comments
 (0)