[Frontend] Fix the TFMA/TFDV viz behavior when caching is used. (kube…

…flow#3228)

* include cached execution

* apply suggestion

* typo fix

* lint

* improve style

* fix typo

* fix

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;
 }
 
 /**