diff --git a/neps/plot/tensorboard_eval.py b/neps/plot/tensorboard_eval.py index 395b5b9a..9259952c 100644 --- a/neps/plot/tensorboard_eval.py +++ b/neps/plot/tensorboard_eval.py @@ -91,7 +91,7 @@ def _initiate_internal_configurations() -> None: # on this object, OR even just use `get_in_process_trial()` in each call directly. tblogger.config_working_directory = trial.pipeline_dir tblogger.config_previous_directory = trial.disk.previous_pipeline_dir - tblogger.optimizer_dir = trial.disk.optimization_dir + tblogger.optimizer_dir = trial.disk.optimization_dir.parent tblogger.config = trial.config @staticmethod