From 20f68ccd5d80908c1776ca062fafc0ec02f5d731 Mon Sep 17 00:00:00 2001 From: Vladimir Lipkin Date: Fri, 13 Dec 2024 18:26:01 +0300 Subject: [PATCH] Decrease default poll_interval and poll_timeout for runs --- src/yandex_cloud_ml_sdk/_runs/run.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/yandex_cloud_ml_sdk/_runs/run.py b/src/yandex_cloud_ml_sdk/_runs/run.py index 6e2ed5e..1d20a2e 100644 --- a/src/yandex_cloud_ml_sdk/_runs/run.py +++ b/src/yandex_cloud_ml_sdk/_runs/run.py @@ -122,8 +122,8 @@ async def wait( self, *, timeout: float = 60, - poll_timeout: int = 3600, - poll_interval: float = 10, + poll_timeout: int = 300, + poll_interval: float = 0.5, ) -> RunResult: return await self._wait( timeout=timeout, @@ -163,8 +163,8 @@ def wait( self, *, timeout: float = 60, - poll_timeout: int = 3600, - poll_interval: float = 10, + poll_timeout: int = 300, + poll_interval: float = 0.5, ) -> RunResult: # NB: mypy can't unterstand normally __wait return type and thinks its ResultTypeT return self.__wait( # type: ignore[return-value]