Embedded Single Card: get loadable time series selector (#6512)

Create a separate selector for the internal single card to fetch time series data with a given card Id.

Author: bhuang1323

tensorboard/webapp/metrics/store/metrics_selectors.ts

@@ -113,20 +113,30 @@ export const getCardLoadState = createSelector(
   }
 );
 
+export const getLoadableTimeSeries = memoize((cardMetadata: CardMetadata) => {
+  return createSelector(
+    (state: MetricsState): MetricsState => state,
+    (state: MetricsState): DeepReadonly<RunToSeries> | null => {
+      const {plugin, tag, sample} = cardMetadata;
+      const loadable = storeUtils.getTimeSeriesLoadable(
+        state.timeSeriesData,
+        plugin,
+        tag,
+        sample
+      );
+      return loadable ? loadable.runToSeries : null;
+    }
+  );
+});
+
 export const getCardTimeSeries = createSelector(
   selectMetricsState,
   (state: MetricsState, cardId: CardId): DeepReadonly<RunToSeries> | null => {
     if (!state.cardMetadataMap.hasOwnProperty(cardId)) {
       return null;
     }
-    const {plugin, tag, sample} = state.cardMetadataMap[cardId];
-    const loadable = storeUtils.getTimeSeriesLoadable(
-      state.timeSeriesData,
-      plugin,
-      tag,
-      sample
-    );
-    return loadable ? loadable.runToSeries : null;
+
+    return getLoadableTimeSeries(state.cardMetadataMap[cardId])(state);
   }
 );
 

tensorboard/webapp/metrics/store/metrics_selectors_test.ts

@@ -404,6 +404,54 @@ describe('metrics selectors', () => {
     });
   });
 
+  describe('getLoadableTimeSeries', () => {
+    it('getLoadableTimeSeries', () => {
+      const sampleScalarRunToSeries = {
+        run1: createScalarStepData(),
+        run2: createScalarStepData(),
+      };
+      const state = buildMetricsState({
+        timeSeriesData: {
+          ...createTimeSeriesData(),
+          scalars: {
+            tagA: {
+              runToLoadState: {
+                run1: DataLoadState.LOADED,
+                run2: DataLoadState.LOADED,
+              },
+              runToSeries: sampleScalarRunToSeries,
+            },
+          },
+        },
+      });
+
+      expect(
+        selectors.getLoadableTimeSeries({
+          plugin: PluginType.SCALARS,
+          tag: 'tagA',
+          runId: null,
+        })(state)
+      ).toEqual(sampleScalarRunToSeries);
+    });
+
+    it('returns null when plugin data does not contain tag', () => {
+      const state = buildMetricsState({
+        timeSeriesData: {
+          ...createTimeSeriesData(),
+          scalars: {},
+        },
+      });
+
+      expect(
+        selectors.getLoadableTimeSeries({
+          plugin: PluginType.SCALARS,
+          tag: 'tagA',
+          runId: null,
+        })(state)
+      ).toBeNull();
+    });
+  });
+
   describe('getCardStepIndex', () => {
     it('returns null if no card exists', () => {
       selectors.getCardStepIndexMetaData.release();