c4patino/slurm-tuner
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update slurm_handler.py

Browse source
This commit is contained in:
Ceferino Patino 2024-10-10 22:22:15 -05:00 • committed by GitHub
commit 2bb95ce6ba
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
hyperparameter_tuner/slurm_handler.py
View file

@ -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):