diff --git a/src/rubrix/server/apis/v0/models/text_classification.py b/src/rubrix/server/apis/v0/models/text_classification.py index 2d65bde003..006d4efde5 100644 --- 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)