diff --git a/frontend/src/lib/OutputArtifactLoader.ts b/frontend/src/lib/OutputArtifactLoader.ts index da2f2796114d..875f84e641ef 100644 --- a/frontend/src/lib/OutputArtifactLoader.ts +++ b/frontend/src/lib/OutputArtifactLoader.ts @@ -472,10 +472,12 @@ async function getExecutionInContextWithPodName( return undefined; // Not found, this is expected to happen normally when there's no mlmd data. } const state = foundExecution.getPropertiesMap().get('state'); - if (!state || state.getStringValue() !== 'complete') { - return undefined; // Execution doesn't have a valid state, or it has not finished. + // Both complete and cached executions are considered valid. + if (state && ['complete', 'cached'].includes(state.getStringValue())) { + return foundExecution; } - return foundExecution; + // No valid execution found. + return undefined; } /**