Ensure docker args are passed for CPU pods too (#470)

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.

Author: airhorns

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: