From 0514b570e20b0fd41de436e3683a0bed1102f1f3 Mon Sep 17 00:00:00 2001 From: Harry Brundage Date: Thu, 11 Dec 2025 08:32:51 -0500 Subject: [PATCH] Ensure docker args are passed for CPU pods too This fixes a bug where calls to create CPU-only pods didn't pass the given docker args down to the GraphQL mutation. This variable is supported by both mutations, but the python argument was only being passed to the mutation in the GPU-pod case. --- runpod/api/mutations/pods.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/runpod/api/mutations/pods.py b/runpod/api/mutations/pods.py index f3be747b..48409a60 100644 --- a/runpod/api/mutations/pods.py +++ b/runpod/api/mutations/pods.py @@ -88,8 +88,7 @@ def generate_pod_deployment_mutation( input_fields.append(f"minVcpuCount: {min_vcpu_count}") if min_memory_in_gb is not None: input_fields.append(f"minMemoryInGb: {min_memory_in_gb}") - if docker_args is not None: - input_fields.append(f'dockerArgs: "{docker_args}"') + if allowed_cuda_versions is not None: cuda_versions = ", ".join(f'"{v}"' for v in allowed_cuda_versions) input_fields.append(f"allowedCudaVersions: [{cuda_versions}]") @@ -106,6 +105,8 @@ def generate_pod_deployment_mutation( else: input_fields.append("dataCenterId: null") + if docker_args is not None: + input_fields.append(f'dockerArgs: "{docker_args}"') if country_code is not None: input_fields.append(f'countryCode: "{country_code}"') if container_disk_in_gb is not None: