Fix for setting task info OccurredAt when not set. (flyteorg#168)

eapolinario · Mar 29, 2021 · 19e65e1 · 19e65e1
1 parent f406cbb
commit 19e65e1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/flyteplugins/go/tasks/pluginmachinery/core/phase.go b/flyteplugins/go/tasks/pluginmachinery/core/phase.go
@@ -131,10 +131,11 @@ var PhaseInfoUndefined = PhaseInfo{phase: PhaseUndefined}
 
 func phaseInfo(p Phase, v uint32, err *core.ExecutionError, info *TaskInfo) PhaseInfo {
 	if info == nil {
+		info = &TaskInfo{}
+	}
+	if info.OccurredAt == nil {
 		t := time.Now()
-		info = &TaskInfo{
-			OccurredAt: &t,
-		}
+		info.OccurredAt = &t
 	}
 	return PhaseInfo{
 		phase:   p,