fix: select metric based on config search metric then fallback to fir…

…st available metric (#1162)
determined-ai · Aug 31, 2020 · c11eda5 · c11eda5
1 parent c9531f0
commit c11eda5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/webui/react/src/pages/TrialDetails/TrialChart.tsx b/webui/react/src/pages/TrialDetails/TrialChart.tsx
@@ -13,9 +13,11 @@ interface Props {
 }
 
 const TrialChart: React.FC<Props> = ({ metricNames, validationMetric, ...props }: Props) => {
-  const [ metric, setMetric ] = useState<MetricName | undefined>(
-    validationMetric ? { name: validationMetric, type: MetricType.Validation } : undefined,
-  );
+  const defaultMetric = metricNames.find(metricName => {
+    return metricName.name === validationMetric && metricName.type === MetricType.Validation;
+  });
+  const fallbackMetric = metricNames && metricNames.length !== 0 ? metricNames[0] : undefined;
+  const [ metric, setMetric ] = useState<MetricName | undefined>(defaultMetric || fallbackMetric);
 
   const data: Partial<PlotData>[] = useMemo(() => {
     const textData: string[] = [];