def generate_gres_conf_script()

in slurm-to-batch/convert_to_batch_job.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

def generate_gres_conf_script(slurm_config: SlurmJobConfig):
    gres_conf_script = f"""#!/bin/bash

# Script to configure Slurm's GPU resources in gres.conf

cat <<EOF > /usr/local/etc/slurm/gres.conf
# Define GPU resources
AutoDetect=nvml
"""
    if slurm_config.gpu_per_node > 0 and slurm_config.gpu_type not in ["None", None, ""]:
        for i in range(slurm_config.gpu_per_node):
            gres_conf_script += f"Name=gpu Type={slurm_config.gpu_type} File=/dev/nvidia{i}\n"
    else:
        for i in range(slurm_config.gpu_per_node):
            gres_conf_script += f"Name=gpu File=/dev/nvidia{i}\n"

    gres_conf_script += "EOF\n"
    return gres_conf_script