Skip to content

Commit 88eabce

Browse files
authored
Change dataset & batch timeouts (#92)
1 parent e0ce8bf commit 88eabce

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/yandex_cloud_ml_sdk/_datasets/uploaders.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
logger = get_logger(__name__)
1818

19-
DEFAULT_CHUNK_SIZE = 500 * 1024 ** 2
19+
DEFAULT_CHUNK_SIZE = 100 * 1024 ** 2 # 100 Mb
2020

2121
MAX_CHUNK_SIZE = 5 * 1024 ** 3 # 5 GB
2222
MAX_CHUNK_SIZE_PRETTY = '5GB'

src/yandex_cloud_ml_sdk/_types/batch/operation.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ def __init__(
4141
transformer=self._result_transformer,
4242
service_name='ai-foundation-models',
4343
initial_operation=initial_operation,
44+
default_poll_timeout=60 * 60 * 72, # 72h
4445
)
4546

4647
# NB: I don't want to make parent operation class Generic[MetadataTypeT] just to

0 commit comments

Comments
 (0)