From a21327a22b83e2d02c7c25ab3c629900bf794d03 Mon Sep 17 00:00:00 2001
From: Gaurav Gupta <gaugup@microsoft.com>
Date: Mon, 30 Oct 2023 13:26:03 -0700
Subject: [PATCH] Refer to predicted value from Counterfactual data

Signed-off-by: Gaurav Gupta <gaugup@microsoft.com>
---
 libs/counterfactuals/src/util/getOriginalData.ts | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/libs/counterfactuals/src/util/getOriginalData.ts b/libs/counterfactuals/src/util/getOriginalData.ts
index a21d4945d0..afacb11365 100644
--- a/libs/counterfactuals/src/util/getOriginalData.ts
+++ b/libs/counterfactuals/src/util/getOriginalData.ts
@@ -4,7 +4,6 @@
 import {
   ICounterfactualData,
   IDataset,
-  ifEnableLargeData,
   JointDataset
 } from "@responsible-ai/core-ui";
 import { localization } from "@responsible-ai/localization";
@@ -21,7 +20,7 @@ export function getOriginalData(
       index
     )
   };
-  if (ifEnableLargeData(dataset) && counterfactualData) {
+  if (counterfactualData) {
     const featureNames = counterfactualData.feature_names_including_target;
     const dataPoint = counterfactualData.test_data[0][0];
     featureNames.forEach((f, index) => {