From c4412824e453b9f7f231f09cee8101bc3bc51c04 Mon Sep 17 00:00:00 2001
From: tongy-msft <91754176+tongyu-microsoft@users.noreply.github.com>
Date: Wed, 12 Jul 2023 16:01:28 -0700
Subject: [PATCH] Update error_analysis_manager.py

---
 .../responsibleai_text/managers/error_analysis_manager.py       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/responsibleai_text/responsibleai_text/managers/error_analysis_manager.py b/responsibleai_text/responsibleai_text/managers/error_analysis_manager.py
index 9db9e27ced..2c1fdb824b 100644
--- a/responsibleai_text/responsibleai_text/managers/error_analysis_manager.py
+++ b/responsibleai_text/responsibleai_text/managers/error_analysis_manager.py
@@ -105,7 +105,7 @@ def predict(self, X):
         predictions = self.predictions[index]
         if self.task_type == ModelTask.MULTILABEL_TEXT_CLASSIFICATION:
             return predictions
-        if self.classes is not None:
+        if self.classes is not None and isinstance(predictions[0], int):
             predictions = [self.classes[y] for y in predictions]
         return predictions