From 68db6367b7357dfb4246c47a949410f752cbc204 Mon Sep 17 00:00:00 2001 From: Tatiana Al-Chueyr Date: Thu, 19 Mar 2026 15:51:46 +0000 Subject: [PATCH] Remove timeout override in Cosmos watcher sensors --- cosmos/operators/_watcher/base.py | 5 ----- cosmos/operators/watcher.py | 5 ----- 2 files changed, 10 deletions(-) diff --git a/cosmos/operators/_watcher/base.py b/cosmos/operators/_watcher/base.py index 36b7e59459..217326ed09 100644 --- a/cosmos/operators/_watcher/base.py +++ b/cosmos/operators/_watcher/base.py @@ -2,7 +2,6 @@ import json import logging -from datetime import timedelta from pathlib import Path from typing import Any @@ -288,8 +287,6 @@ def __init__( profiles_dir: str | None = None, producer_task_id: str = PRODUCER_WATCHER_TASK_ID, poke_interval: int = 10, - timeout: int = 60 * 60, # 1 h safety valve - execution_timeout: timedelta = timedelta(hours=1), deferrable: bool = True, **kwargs: Any, ) -> None: @@ -303,8 +300,6 @@ def __init__( extra_context = kwargs.pop("extra_context") if "extra_context" in kwargs else {} super().__init__( poke_interval=poke_interval, - timeout=timeout, - execution_timeout=execution_timeout, profile_config=profile_config, project_dir=project_dir, profiles_dir=profiles_dir, diff --git a/cosmos/operators/watcher.py b/cosmos/operators/watcher.py index 98b11a59d6..bd0301990b 100644 --- a/cosmos/operators/watcher.py +++ b/cosmos/operators/watcher.py @@ -5,7 +5,6 @@ import json import zlib from collections.abc import Callable, Sequence -from datetime import timedelta from typing import TYPE_CHECKING, Any from airflow.exceptions import AirflowException @@ -254,15 +253,11 @@ def __init__( profiles_dir: str | None = None, producer_task_id: str = PRODUCER_WATCHER_TASK_ID, poke_interval: int = 10, - timeout: int = 60 * 60, # 1 h safety valve - execution_timeout: timedelta = timedelta(hours=1), deferrable: bool = True, **kwargs: Any, ) -> None: super().__init__( poke_interval=poke_interval, - timeout=timeout, - execution_timeout=execution_timeout, profile_config=profile_config, project_dir=project_dir, profiles_dir=profiles_dir,