Dropdown option fix (#2505)

* dropdown option fix * auto lint fixes
microsoft · Jan 25, 2024 · 7ff3476 · 7ff3476
1 parent 6973ee4
commit 7ff3476
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/libs/interpret-vision/src/lib/VisionExplanationDashboard/Controls/FlyoutObjectDetection.tsx b/libs/interpret-vision/src/lib/VisionExplanationDashboard/Controls/FlyoutObjectDetection.tsx
@@ -57,7 +57,8 @@ export class FlyoutObjectDetection extends React.Component<
     const selectableObjectIndexes =
       FlyoutStyles.generateSelectableObjectDetectionIndexes(
         localization.InterpretVision.Dashboard.prefix,
-        item
+        item,
+        this.props.dataset.class_names
       );
     this.setState({ item, metadata, selectableObjectIndexes });
   }
@@ -75,7 +76,8 @@ export class FlyoutObjectDetection extends React.Component<
       const selectableObjectIndexes =
         FlyoutStyles.generateSelectableObjectDetectionIndexes(
           localization.InterpretVision.Dashboard.prefix,
-          item
+          item,
+          this.props.dataset.class_names
         );
       this.setState({
         item: this.props.item,

diff --git a/libs/interpret-vision/src/lib/VisionExplanationDashboard/utils/FlyoutUtils.tsx b/libs/interpret-vision/src/lib/VisionExplanationDashboard/utils/FlyoutUtils.tsx
@@ -34,13 +34,14 @@ export function onRenderCell(
 
 export function generateSelectableObjectDetectionIndexes(
   prefix: string,
-  item: IVisionListItem | undefined
+  item: IVisionListItem | undefined,
+  classNames: string[] | undefined
 ): IComboBoxOption[] {
   const temp = item?.odPredictedY;
   const selectableObjectIndexes: IComboBoxOption[] = [];
-  if (temp) {
+  if (temp && classNames) {
     for (let i = 0; i < Object.values(temp).length; i++) {
-      const className = item?.predictedY[i];
+      const className = classNames[temp[i][0] - 1];
       selectableObjectIndexes.push({
         key: prefix + String(i),
         text: String(i) + String(": ") + className