Add error checking to container cleanup

NVIDIA · Aug 18, 2022 · 26ed077 · 26ed077
1 parent 3569a94
commit 26ed077
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/container/docker.go b/internal/container/docker.go
@@ -64,14 +64,20 @@ func (c *DockerContainer) Start() error {
 			return err
 		}
 		if info.State.Status == "exited" {
-			c.Remove()
+			err := c.Remove()
+			if err != nil {
+				return err
+			}
 			return errors.New("container failed to start")
 		}
 		if info.State.Running {
 			break
 		}
 		if time.Since(startTime) > (time.Second * 10) {
-			c.Remove()
+			err := c.Remove()
+			if err != nil {
+				return err
+			}
 			return errors.New("container failed to start after 10 seconds")
 		}
 		time.Sleep(time.Second)