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: