File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change 44import logging
55import math
66import multiprocessing
7+ import os
78import time
89import traceback
910import typing
2526from autoPyTorch .evaluation .utils import empty_queue , extract_learning_curve , read_queue
2627from autoPyTorch .pipeline .components .training .metrics .base import autoPyTorchMetric
2728from autoPyTorch .utils .backend import Backend
29+ from autoPyTorch .utils .common import replace_string_bool_to_bool
2830from autoPyTorch .utils .hyperparameter_search_space_update import HyperparameterSearchSpaceUpdates
2931from 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 :
You can’t perform that action at this time.
0 commit comments