diff --git a/ray_provider/operators/ray.py b/ray_provider/operators/ray.py index ea33b89..1ef859a 100644 --- a/ray_provider/operators/ray.py +++ b/ray_provider/operators/ray.py @@ -281,6 +281,7 @@ def execute(self, context: Context) -> str: current_status = self.hook.get_ray_job_status(self.dashboard_url, self.job_id) self.log.info(f"Current job status for {self.job_id} is: {current_status}") + job_timeout_seconds = timedelta(seconds=0) if isinstance(self.job_timeout_seconds, timedelta): job_timeout_seconds = self.job_timeout_seconds elif self.job_timeout_seconds is not None: