diff --git a/src/dstack/_internal/server/services/runner/client.py b/src/dstack/_internal/server/services/runner/client.py index d8aae611c..e205990aa 100644 --- a/src/dstack/_internal/server/services/runner/client.py +++ b/src/dstack/_internal/server/services/runner/client.py @@ -32,6 +32,7 @@ from dstack._internal.utils.logging import get_logger REQUEST_TIMEOUT = 9 +UPLOAD_CODE_REQUEST_TIMEOUT = 60 logger = get_logger(__name__) @@ -109,7 +110,9 @@ def submit_job( resp.raise_for_status() def upload_code(self, file: Union[BinaryIO, bytes]): - resp = requests.post(self._url("/api/upload_code"), data=file, timeout=REQUEST_TIMEOUT) + resp = requests.post( + self._url("/api/upload_code"), data=file, timeout=UPLOAD_CODE_REQUEST_TIMEOUT + ) resp.raise_for_status() def run_job(self):