Skip to content

Commit

Permalink
change print statements to log statements
Browse files Browse the repository at this point in the history
  • Loading branch information
bgunnar5 committed Oct 23, 2023
1 parent 87daaef commit 85855b3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions merlin/study/batch.py
Original file line number Diff line number Diff line change
Expand Up @@ -335,15 +335,15 @@ def construct_worker_launch_command(parsed_batch: Dict, nodes: int) -> str:
scheduler_legend: Dict = construct_scheduler_legend(parsed_batch, nodes)
workload_manager: str = get_batch_type(scheduler_legend)

print(f"parsed_batch: {parsed_batch}")
LOG.debug(f"parsed_batch: {parsed_batch}")

if parsed_batch["btype"] == "pbs" and workload_manager == parsed_batch["btype"]:
raise TypeError("The PBS scheduler is only enabled for 'batch: flux' type")

if parsed_batch["btype"] == "slurm" and workload_manager not in ("lsf", "flux", "pbs"):
workload_manager = "slurm"

print(f"workload_manager: {workload_manager}")
LOG.debug(f"workload_manager: {workload_manager}")

try:
launch_command = scheduler_legend[workload_manager]["launch"]
Expand Down

0 comments on commit 85855b3

Please sign in to comment.