fix: keep allowed_cuda_versions as string (#421)

* fix: keep `allowed_cuda_versions` as string
* fix: do not include when empty or null

Author: deanq

runpod/api/ctl_commands.py

@@ -316,7 +316,7 @@ def create_endpoint(
     workers_min: int = 0,
     workers_max: int = 3,
     flashboot=False,
-    allowed_cuda_versions: list = None,
+    allowed_cuda_versions: str = None,
     gpu_count: int = 1,
 ):
     """

runpod/api/mutations/endpoints.py

@@ -15,7 +15,7 @@ def generate_endpoint_mutation(
     workers_min: int = 0,
     workers_max: int = 3,
     flashboot=False,
-    allowed_cuda_versions: list = None,
+    allowed_cuda_versions: str = None,
     gpu_count: int = None,
 ):
     """Generate a string for a GraphQL mutation to create a new endpoint."""
@@ -47,8 +47,7 @@ def generate_endpoint_mutation(
     input_fields.append(f"workersMax: {workers_max}")
 
     if allowed_cuda_versions:
-        cuda_versions = ", ".join(f'"{v}"' for v in allowed_cuda_versions)
-        input_fields.append(f"allowedCudaVersions: [{cuda_versions}]")
+        input_fields.append(f"allowedCudaVersions: {allowed_cuda_versions}")
 
     if gpu_count is not None:
         input_fields.append(f"gpuCount: {gpu_count}")