diff --git a/task-sdk/src/airflow/sdk/execution_time/cache.py b/task-sdk/src/airflow/sdk/execution_time/cache.py index f3a0593685933..c736bd8e1685e 100644 --- a/task-sdk/src/airflow/sdk/execution_time/cache.py +++ b/task-sdk/src/airflow/sdk/execution_time/cache.py @@ -20,8 +20,6 @@ import datetime import multiprocessing -from airflow.sdk import timezone - class SecretCache: """A static class to manage the global secret cache.""" @@ -35,10 +33,14 @@ class NotPresentException(Exception): class _CacheValue: def __init__(self, value: str | None) -> None: + from airflow.sdk import timezone + self.value = value self.date = timezone.utcnow() def is_expired(self, ttl: datetime.timedelta) -> bool: + from airflow.sdk import timezone + return timezone.utcnow() - self.date > ttl _VARIABLE_PREFIX = "__v_"