fix: TaskWorker needs to implement only consume_work

provos · Aug 31, 2024 · 99e52e5 · 99e52e5
1 parent 1586c3a
commit 99e52e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/planai/task.py b/src/planai/task.py
@@ -102,7 +102,7 @@ class TaskWorker(BaseModel, ABC):
         _consumers (Dict[Type["TaskWorker"], "TaskWorker"]): A private attribute to store registered consumers.
 
     Note:
-        Any subclass of TaskWorker must implement both consume_work and publish_work methods.
+        Any subclass of TaskWorker must implement consume_work.
     """
 
     output_types: Set[Type[TaskWorkItem]] = Field(