Entity-related typing changes

Author: andystaples

durabletask/client.py

@@ -230,7 +230,10 @@ def purge_orchestration(self, instance_id: str, recursive: bool = True):
         self._logger.info(f"Purging instance '{instance_id}'.")
         self._stub.PurgeInstances(req)
 
-    def signal_entity(self, entity_instance_id: EntityInstanceId, operation_name: str, input: Optional[Any] = None):
+    def signal_entity(self,
+                      entity_instance_id: EntityInstanceId,
+                      operation_name: str,
+                      input: Optional[Any] = None) -> None:
         req = pb.SignalEntityRequest(
             instanceId=str(entity_instance_id),
             name=operation_name,

durabletask/task.py

@@ -142,7 +142,7 @@ def call_activity(self, activity: Union[Activity[TInput, TOutput], str], *,
     def call_entity(self,
                     entity: EntityInstanceId,
                     operation: str,
-                    input: Optional[TInput] = None) -> CompletableTask:
+                    input: Optional[TInput] = None) -> CompletableTask[Any]:
         """Schedule entity function for execution.
 
         Parameters

durabletask/worker.py

@@ -1047,7 +1047,7 @@ def call_entity(
             entity: EntityInstanceId,
             operation: str,
             input: Optional[TInput] = None,
-    ) -> task.CompletableTask:
+    ) -> task.CompletableTask[Any]:
         id = self.next_sequence_number()
 
         self.call_entity_function_helper(