Skip to content

Commit ce4707d

Browse files
committed
Allow the GPU model to be optional without causing errors
1 parent b538a6f commit ce4707d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

workflow/Snakefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ rule structure_inference:
243243
slurm_partition=(DEFAULT_SLURM_PARTITION or "gpu"),
244244
qos=DEFAULT_SLURM_QOS,
245245
gpu=DEFAULT_STRUCTURE_INFERENCE_GPUS,
246-
gpu_model=DEFAULT_STRUCTURE_INFERENCE_GPU_MODEL,
246+
**({"gpu_model": DEFAULT_STRUCTURE_INFERENCE_GPU_MODEL} if DEFAULT_STRUCTURE_INFERENCE_GPU_MODEL is not None else {}),
247247
tasks_per_gpu=DEFAULT_STRUCTURE_INFERENCE_TASKS_PER_GPU,
248248
**linear_resources(
249249
mem_fn=lambda wc, attempt: config.get("structure_inference_ram_bytes", 32000)

0 commit comments

Comments
 (0)