Linting

Author: andystaples

durabletask/entities/durable_entity.py

@@ -11,13 +11,16 @@ def _initialize_entity_context(self, context: EntityContext):
         self.entity_context = context
 
     @overload
-    def get_state(self, intended_type: Type[TState], default: TState) -> TState: ...
+    def get_state(self, intended_type: Type[TState], default: TState) -> TState:
+        ...
 
     @overload
-    def get_state(self, intended_type: Type[TState]) -> Optional[TState]: ...
+    def get_state(self, intended_type: Type[TState]) -> Optional[TState]:
+        ...
 
     @overload
-    def get_state(self, intended_type: None = None, default: Any = None) -> Any: ...
+    def get_state(self, intended_type: None = None, default: Any = None) -> Any:
+        ...
 
     def get_state(self, intended_type: Optional[Type[TState]] = None, default: Optional[TState] = None) -> Optional[TState] | Any:
         return self.entity_context.get_state(intended_type, default)

durabletask/internal/entity_state_shim.py

@@ -14,13 +14,16 @@ def __init__(self, start_state):
         self._actions_checkpoint_state: int = 0
 
     @overload
-    def get_state(self, intended_type: Type[TState], default: TState) -> TState: ...
+    def get_state(self, intended_type: Type[TState], default: TState) -> TState:
+        ...
 
     @overload
-    def get_state(self, intended_type: Type[TState]) -> Optional[TState]: ...
+    def get_state(self, intended_type: Type[TState]) -> Optional[TState]:
+        ...
 
     @overload
-    def get_state(self, intended_type: None = None, default: Any = None) -> Any: ...
+    def get_state(self, intended_type: None = None, default: Any = None) -> Any:
+        ...
 
     def get_state(self, intended_type: Optional[Type[TState]] = None, default: Optional[TState] = None) -> Optional[TState] | Any:
         if self._current_state is None and default is not None:

durabletask/task.py

@@ -549,13 +549,16 @@ def operation(self) -> str:
         return self._operation
 
     @overload
-    def get_state(self, intended_type: Type[TState], default: TState) -> TState: ...
+    def get_state(self, intended_type: Type[TState], default: TState) -> TState:
+        ...
 
     @overload
-    def get_state(self, intended_type: Type[TState]) -> Optional[TState]: ...
+    def get_state(self, intended_type: Type[TState]) -> Optional[TState]:
+        ...
 
     @overload
-    def get_state(self, intended_type: None = None, default: Any = None) -> Any: ...
+    def get_state(self, intended_type: None = None, default: Any = None) -> Any:
+        ...
 
     def get_state(self, intended_type: Optional[Type[TState]] = None, default: Optional[TState] = None) -> Optional[TState] | Any:
         return self._state.get_state(intended_type, default)