change stategy "BOTH" to "CONSENSUS_WITH_THRESHOLD"

enoch3712 · Jun 19, 2024 · 842a9dc · 842a9dc
1 parent 508e7e3
commit 842a9dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extract_thinker/process.py b/extract_thinker/process.py
@@ -17,7 +17,7 @@
 class ClassificationStrategy(Enum):
     CONSENSUS = "consensus"
     HIGHER_ORDER = "higher_order"
-    BOTH = "both"
+    CONSENSUS_WITH_THRESHOLD = "both"
 
 
 class Process:
@@ -73,7 +73,7 @@ async def classify_async(self, file: str, classifications, strategy: Classificat
         elif strategy == ClassificationStrategy.HIGHER_ORDER:
             # Pick the result with the highest confidence
             return max(group_classifications, key=lambda c: c.confidence)
-        elif strategy == ClassificationStrategy.BOTH:
+        elif strategy == ClassificationStrategy.CONSENSUS_WITH_THRESHOLD:
             if len(set(group_classifications)) == 1:
                 maxResult = max(group_classifications, key=lambda c: c.confidence)
                 if maxResult.confidence >= threshold: