|
49 | 49 | from .task_shared_volume import TaskSharedVolumes |
50 | 50 |
|
51 | 51 | _logger = logging.getLogger(__name__) |
52 | | -CONTAINER_WAIT_TIME_SECS = 2 |
53 | | -MAX_LOGGED_FILE_CHARS = 40 |
| 52 | +_CONTAINER_WAIT_TIME_SECS = 2 |
| 53 | +_MAX_LOGGED_FILE_CHARS = 40 |
54 | 54 |
|
55 | 55 |
|
56 | 56 | @dataclass(kw_only=True, frozen=True, slots=True) |
@@ -154,7 +154,10 @@ async def _retrieve_output_data( |
154 | 154 |
|
155 | 155 | src_path = task_volumes.outputs_folder / output_params.file_mapping |
156 | 156 | await log_partial_file_content( |
157 | | - src_path, _logger, MAX_LOGGED_FILE_CHARS |
| 157 | + src_path, |
| 158 | + logger=_logger, |
| 159 | + log_level=logging.DEBUG, |
| 160 | + max_chars=_MAX_LOGGED_FILE_CHARS, |
158 | 161 | ) |
159 | 162 | upload_tasks.append( |
160 | 163 | push_file_to_remote( |
@@ -271,7 +274,7 @@ async def run(self, command: list[str]) -> TaskOutputData: |
271 | 274 | ) |
272 | 275 | # wait until the container finished, either success or fail or timeout |
273 | 276 | while (container_data := await container.show())["State"]["Running"]: |
274 | | - await asyncio.sleep(CONTAINER_WAIT_TIME_SECS) |
| 277 | + await asyncio.sleep(_CONTAINER_WAIT_TIME_SECS) |
275 | 278 | if container_data["State"]["ExitCode"] > os.EX_OK: |
276 | 279 | raise ServiceRuntimeError( |
277 | 280 | service_key=self.task_parameters.image, |
|
0 commit comments