fix multiple inference index error

GT4SD · Jun 7, 2023 · cf089c0 · cf089c0
1 parent 816e763
commit cf089c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/zberta/intent_discovery/zberta.py b/src/zberta/intent_discovery/zberta.py
@@ -33,7 +33,7 @@ def find_new_intents(self, utterances: List = None, lang='en'):
                 classes = unknown_intents_set([utterances[i]], lang)
             else:
                 classes = self.z_classes
-            for class_set in classes[i]:
+            for class_set in classes[0]:
                 temp = class_set.replace('-', ' ').replace('None', '')
                 if temp not in pred_classes and len(temp) > 1:
                     pred_classes.append(temp)