Skip to content

Commit

Permalink
[FIX] flake
Browse files Browse the repository at this point in the history
  • Loading branch information
franchuterivera committed Apr 21, 2021
1 parent a4853c0 commit 2046154
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion autoPyTorch/evaluation/tae.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import logging
import math
import multiprocessing
import os
import time
import traceback
import typing
Expand All @@ -25,6 +26,7 @@
from autoPyTorch.evaluation.utils import empty_queue, extract_learning_curve, read_queue
from autoPyTorch.pipeline.components.training.metrics.base import autoPyTorchMetric
from autoPyTorch.utils.backend import Backend
from autoPyTorch.utils.common import replace_string_bool_to_bool
from autoPyTorch.utils.hyperparameter_search_space_update import HyperparameterSearchSpaceUpdates
from autoPyTorch.utils.logging_ import PicklableClientLogger, get_named_client_logger

Expand Down Expand Up @@ -144,7 +146,12 @@ def __init__(
self.exclude = exclude
self.disable_file_output = disable_file_output
self.init_params = init_params
self.pipeline_config = pipeline_config
self.pipeline_config: typing.Dict[str, typing.Union[int, str, float]] = dict()
if pipeline_config is None:
pipeline_config = replace_string_bool_to_bool(json.load(open(
os.path.join(os.path.dirname(__file__), '../configs/default_pipeline_options.json'))))
self.pipeline_config.update(pipeline_config)

self.budget_type = pipeline_config['budget_type'] if pipeline_config is not None else budget_type
self.logger_port = logger_port
if self.logger_port is None:
Expand Down

0 comments on commit 2046154

Please sign in to comment.