Use utils to get logs path

Author: polyaxon-ci

cli/polyaxon/_cli/operations.py

@@ -59,6 +59,7 @@
     PolyaxonShouldExitError,
 )
 from polyaxon.logger import clean_outputs
+from traceml.events import get_logs_path
 
 DEFAULT_EXCLUDE = [
     "owner",
@@ -1449,7 +1450,7 @@ def logs(ctx, project, uid, follow, hide_time, all_containers, all_info, offline
         offline_path = ctx_paths.get_offline_path(
             entity_value=uid, entity_kind=V1ProjectFeature.RUNTIME, path=path
         )
-        logs_path = "{}/plxlogs".format(offline_path)
+        logs_path = get_logs_path(run_path=offline_path, full_path=False)
         offline_path = "{}/{}".format(offline_path, ctx_paths.CONTEXT_LOCAL_RUN)
         if not os.path.exists(offline_path):
             Printer.error(

cli/polyaxon/_sidecar/container/__init__.py

@@ -129,7 +129,7 @@ async def monitor():
     while is_running:
         await asyncio.sleep(sleep_interval)
         if retry:
-            await asyncio.sleep(retry ** 2)
+            await asyncio.sleep(retry**2)
         try:
             is_running = await k8s_manager.is_pod_running(pod_id, container_id)
         except ApiException as e:

cli/polyaxon/_sidecar/container/monitors/logs.py

@@ -11,6 +11,7 @@
 from polyaxon._k8s.logging.async_monitor import query_k8s_pod_logs
 from polyaxon._k8s.manager.async_manager import AsyncK8sManager
 from traceml.logging import V1Logs
+from traceml.events import get_logs_path
 
 
 async def sync_logs(
@@ -20,9 +21,6 @@ async def sync_logs(
     last_time: Optional[datetime.datetime],
     stream: bool = False,
 ) -> Optional[datetime.datetime]:
-    path_from = ctx_paths.CONTEXT_MOUNT_ARTIFACTS_FORMAT.format(run_uuid)
-    path_from = "{}/plxlogs".format(path_from)
-
     logs, last_time = await query_k8s_pod_logs(
         k8s_manager=k8s_manager,
         pod=pod,
@@ -32,7 +30,10 @@ async def sync_logs(
     if not logs:
         return last_time
 
-    path_from = "{}/{}.jsonl".format(path_from, pod.metadata.name)
+    path_from = get_logs_path(
+        run_path=ctx_paths.CONTEXT_MOUNT_ARTIFACTS_FORMAT.format(run_uuid),
+        filename=pod.metadata.name,
+    )
     check_or_create_path(path_from, is_dir=False)
     async with aiofiles.open(path_from, "a") as outfile:
         _logs = V1Logs.construct(logs=logs)