Fix problem with device request not being list

roboflow · Dec 18, 2023 · 0de3ce2 · 0de3ce2
1 parent f6edb4a
commit 0de3ce2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/inference_cli/lib/container_adapter.py b/inference_cli/lib/container_adapter.py
@@ -84,9 +84,9 @@ def start_inference_container(
     privileged = False
     if "gpu" in image:
         privileged = True
-        device_requests = (
-            [docker.types.DeviceRequest(device_ids=["all"], capabilities=[["gpu"]])],
-        )
+        device_requests = [
+            docker.types.DeviceRequest(device_ids=["all"], capabilities=[["gpu"]])
+        ]
 
     print(f"Starting inference server container...")
     docker_client.containers.run(