Skip to content

Commit 7d715e8

Browse files
committed
Match entity method parameter names
- Ensure consistency in worker.py to task.py
1 parent b4086fd commit 7d715e8

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

durabletask/task.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,8 @@ def call_activity(self, activity: Union[Activity[TInput, TOutput], str], *,
139139
pass
140140

141141
@abstractmethod
142-
def call_entity(self, entity: EntityInstanceId,
142+
def call_entity(self,
143+
entity: EntityInstanceId,
143144
operation: str,
144145
input: Optional[TInput] = None) -> Task:
145146
"""Schedule entity function for execution.

durabletask/worker.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1038,28 +1038,28 @@ def call_activity(
10381038

10391039
def call_entity(
10401040
self,
1041-
entity_id: EntityInstanceId,
1041+
entity: EntityInstanceId,
10421042
operation: str,
10431043
input: Optional[TInput] = None,
10441044
) -> task.Task:
10451045
id = self.next_sequence_number()
10461046

10471047
self.call_entity_function_helper(
1048-
id, entity_id, operation, input=input
1048+
id, entity, operation, input=input
10491049
)
10501050

10511051
return self._pending_tasks.get(id, task.CompletableTask())
10521052

10531053
def signal_entity(
10541054
self,
10551055
entity_id: EntityInstanceId,
1056-
operation: str,
1056+
operation_name: str,
10571057
input: Optional[TInput] = None
10581058
) -> None:
10591059
id = self.next_sequence_number()
10601060

10611061
self.signal_entity_function_helper(
1062-
id, entity_id, operation, input
1062+
id, entity_id, operation_name, input
10631063
)
10641064

10651065
def lock_entities(self, entities: list[EntityInstanceId]) -> task.Task[EntityLock]:

0 commit comments

Comments
 (0)