Merge pull request #12 from gilpasternak/gil/fix_multi_inference

fix multiple inference index error
GT4SD · Jun 13, 2023 · 00d26b1 · 00d26b1
2 parents 816e763 + cf089c0
commit 00d26b1
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)