edits

determined-ai · Nov 17, 2023 · 0b14e65 · 0b14e65
1 parent 4b70c21
commit 0b14e65
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/master/internal/command/command.go b/master/internal/command/command.go
@@ -237,13 +237,13 @@ func (c *Command) OnExit(ae *task.AllocationExited) {
 
 	c.exitStatus = ae
 
-	var sErr status.Status
 	err := c.db.CompleteTask(c.taskID, time.Now().UTC())
 	if err != nil {
-		c.syslog.WithError(err).Error("marking task complete")
+		var sErr status.Status
 		if errors.As(err, &sErr) && sErr.Code() != codes.NotFound {
 			return
 		}
+		c.syslog.WithError(err).Error("marking task complete")
 	}
 
 	if err := user.DeleteSessionByToken(context.TODO(), c.GenericCommandSpec.Base.UserSessionToken); err != nil {