2828from smac .stats .stats import Stats
2929from smac .tae import StatusType
3030
31+ from autoPyTorch .automl_common .common .utils .backend import Backend , create
3132from autoPyTorch .constants import (
3233 REGRESSION_TASKS ,
3334 STRING_TO_OUTPUT_TYPES ,
4546from autoPyTorch .pipeline .components .setup .traditional_ml .classifier_models import get_available_classifiers
4647from autoPyTorch .pipeline .components .training .metrics .base import autoPyTorchMetric
4748from autoPyTorch .pipeline .components .training .metrics .utils import calculate_score , get_metrics
48- from autoPyTorch .utils .backend import Backend , create
4949from autoPyTorch .utils .common import FitRequirement , replace_string_bool_to_bool
5050from autoPyTorch .utils .hyperparameter_search_space_update import HyperparameterSearchSpaceUpdates
5151from autoPyTorch .utils .logging_ import (
@@ -157,6 +157,7 @@ def __init__(
157157 self ._backend = backend
158158 else :
159159 self ._backend = create (
160+ prefix = 'autoPyTorch' ,
160161 temporary_directory = self ._temporary_directory ,
161162 output_directory = self ._output_directory ,
162163 delete_tmp_folder_after_terminate = delete_tmp_folder_after_terminate ,
@@ -776,7 +777,6 @@ def _search(
776777 self
777778
778779 """
779-
780780 if self .task_type != dataset .task_type :
781781 raise ValueError ("Incompatible dataset entered for current task,"
782782 "expected dataset to have task type :{} got "
@@ -794,6 +794,10 @@ def _search(
794794
795795 if self ._logger is None :
796796 self ._logger = self ._get_logger (self .dataset_name )
797+
798+ # Setup the logger for the backend
799+ self ._backend .setup_logger (port = self ._logger_port )
800+
797801 self ._all_supported_metrics = all_supported_metrics
798802 self ._disable_file_output = disable_file_output
799803 self ._memory_limit = memory_limit
0 commit comments