fix: wrong predicted condition

argilla-io · Jun 2, 2022 · ff387a1 · ff387a1
1 parent f15f9bb
commit ff387a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rubrix/server/apis/v0/models/text_classification.py b/src/rubrix/server/apis/v0/models/text_classification.py
@@ -297,7 +297,7 @@ def task(cls) -> TaskType:
 
     @property
     def predicted(self) -> Optional[PredictionStatus]:
-        if self.predicted_by and self.predicted_by:
+        if self.predicted_by and self.annotated_by:
             return (
                 PredictionStatus.OK
                 if set(self.predicted_as) == set(self.annotated_as)