From 2bb95ce6baecf9087eaa8caed84a3cbd8c5c3851 Mon Sep 17 00:00:00 2001 From: Ceferino Patino Date: Thu, 10 Oct 2024 22:22:15 -0500 Subject: [PATCH] Update slurm_handler.py --- hyperparameter_tuner/slurm_handler.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/hyperparameter_tuner/slurm_handler.py b/hyperparameter_tuner/slurm_handler.py index 60ad290..5f5daf2 100644 --- a/hyperparameter_tuner/slurm_handler.py +++ b/hyperparameter_tuner/slurm_handler.py @@ -10,9 +10,7 @@ import pandas as pd from hyperparameter_tuner.loss import Loss -logging.basicConfig(level=logging.INFO) -logger = logging.getLogger(__name__) - +logger = logging.getLogger('main') def create_objective( slurm_script: str, @@ -65,10 +63,10 @@ def create_objective( try: subprocess.run(command, shell=True, check=True) - logger.info('SLURM job submitted with trial ID: {trial_id}') - logger.info('Paramters: {trial_params}') + logger.info(f'SLURM job submitted with trial ID: {trial_id}') + logger.info(f'Paramters: {trial_params}') except subprocess.CalledProcessError: - logger.error('Error submitting SLURM job') + logger.error(f'Error submitting SLURM job') raise while not os.path.isfile(results_path):