Fix tests

jumppad-labs · Apr 5, 2024 · b149d7d · b149d7d
1 parent 45c329e
commit b149d7d
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/pkg/clients/container/docker_tasks.go b/pkg/clients/container/docker_tasks.go
@@ -187,19 +187,20 @@ func (d *DockerTasks) CreateContainer(c *dtypes.Container) (string, error) {
 		}
 
 		hc.Resources = rc
-	}
 
-	// Add GPU details
-	if c.Resources.GPU != nil {
-		hc.DeviceRequests = []container.DeviceRequest{
-			{
-				Driver:       c.Resources.GPU.Driver,
-				DeviceIDs:    c.Resources.GPU.DeviceIDs,
-				Capabilities: [][]string{[]string{"gpu", c.Resources.GPU.Driver, "compute"}},
-			},
+		if c.Resources.GPU != nil {
+			hc.DeviceRequests = []container.DeviceRequest{
+				{
+					Driver:       c.Resources.GPU.Driver,
+					DeviceIDs:    c.Resources.GPU.DeviceIDs,
+					Capabilities: [][]string{[]string{"gpu", c.Resources.GPU.Driver, "compute"}},
+				},
+			}
 		}
 	}
 
+	// Add GPU details
+
 	// by default the container should NOT be attached to a network
 	nc.EndpointsConfig = make(map[string]*network.EndpointSettings)