amoghrajesh commented on code in PR #66859:
URL: https://github.com/apache/airflow/pull/66859#discussion_r3264007437
##########
task-sdk/src/airflow/sdk/execution_time/context.py:
##########
@@ -434,11 +435,25 @@ def get(self, key, default: Any = NOTSET) -> Any:
raise
+@cache
+def _get_worker_state_backend():
Review Comment:
Good catch, made it `BaseStateBackend | None` since the function returns
`None` when no worker backend is configured (callers check `if backend is not
None:`). Using BaseStateBackend alone would be incorrect.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
