polyaxon 2.7.0rc0__py3-none-any.whl → 2.8.0rc0__py3-none-any.whl

polyaxon/_cli/operations.py

@@ -730,7 +730,7 @@ def stop(ctx, project, uid, yes):
         is_cli=True,
     )
     if not yes and not click.confirm(
-        "Are sure you want to stop " "run `{}`".format(run_uuid)
+        "Are sure you want to stop run `{}`".format(run_uuid)
     ):
         Printer.print("Exiting without stopping run.")
         sys.exit(0)
@@ -782,7 +782,7 @@ def skip(ctx, project, uid, yes):
         is_cli=True,
     )
     if not yes and not click.confirm(
-        "Are sure you want to stop " "run `{}`".format(run_uuid)
+        "Are sure you want to stop run `{}`".format(run_uuid)
     ):
         Printer.print("Exiting without stopping run.")
         sys.exit(0)
@@ -1854,9 +1854,17 @@ def artifacts(
 @click.option(
     "--agent", "-ag", type=str, help="Optional, uuid reference of an agent to use."
 )
+@click.option(
+    "--ignore-agent-host",
+    is_flag=True,
+    default=False,
+    help="Optional, to ignore the agent host.",
+)
 @click.pass_context
 @clean_outputs
-def upload(ctx, project, uid, path_from, path_to, sync_failure, agent):
+def upload(
+    ctx, project, uid, path_from, path_to, sync_failure, agent, ignore_agent_host
+):
     """Upload runs' artifacts.
 
     Uses /docs/core/cli/#caching
@@ -1887,7 +1895,11 @@ def upload(ctx, project, uid, path_from, path_to, sync_failure, agent):
         )
         if is_file:
             response = client.upload_artifact(
-                filepath=path_from, path=path_to, overwrite=True, agent=agent
+                filepath=path_from,
+                path=path_to,
+                overwrite=True,
+                agent=agent,
+                ignore_agent_host=ignore_agent_host,
             )
         else:
             response = client.upload_artifacts_dir(
@@ -1896,6 +1908,7 @@ def upload(ctx, project, uid, path_from, path_to, sync_failure, agent):
                 overwrite=True,
                 relative_to=path_from,
                 agent=agent,
+                ignore_agent_host=ignore_agent_host,
             )
     except (
         ApiException,
@@ -2310,11 +2323,17 @@ def pull(
     "--reset-uuid",
     is_flag=True,
     default=False,
-    help="Optional, to ignore the uuid of the local " "run and generate a new uuid.",
+    help="Optional, to ignore the uuid of the local run and generate a new uuid.",
 )
 @click.option(
     "--agent", "-ag", type=str, help="Optional, uuid reference of an agent to use."
 )
+@click.option(
+    "--ignore-agent-host",
+    is_flag=True,
+    default=False,
+    help="Optional, to ignore the agent host.",
+)
 @click.option("--name", "-n", type=str, help="Optional, a new name to set for the run.")
 @click.pass_context
 @clean_outputs
@@ -2329,6 +2348,7 @@ def push(
     reset_project,
     reset_uuid,
     agent,
+    ignore_agent_host,
     name,
 ):
     """Push a local run (or all runs) to a remove server.
@@ -2352,6 +2372,9 @@ def push(
     \b
     $ polyaxon ops push -uid 8aac02e3a62a4f0aaa257c59da5eab80 --reset-project
 
+    \b
+    $ polyaxon ops push -uid 8aac02e3a62a4f0aaa257c59da5eab80 --ignore-agent-host
+
     \b
     $ polyaxon ops push -uid 8aac02e3a62a4f0aaa257c59da5eab80 --reset-project -p send-to-project
     """
@@ -2397,6 +2420,7 @@ def push(
                 upload_artifacts=not no_artifacts,
                 clean=clean,
                 agent=agent,
+                ignore_agent_host=ignore_agent_host,
             )
             Printer.success(
                 f"Finished pushing offline run {uid} to {client.owner}/{client.project}"

polyaxon/_client/run.py

@@ -1517,6 +1517,7 @@ class RunClient(ClientMixin):
         overwrite: bool = True,
         show_progress: bool = True,
         agent: Optional[str] = None,
+        ignore_agent_host: bool = False,
     ):
         """Uploads a single artifact to the run's artifacts store path.
 
@@ -1528,16 +1529,16 @@ class RunClient(ClientMixin):
             overwrite: bool, optional, if the file uploaded should overwrite any previous content.
             show_progress: bool, to show a progress bar.
             agent: str, optional, uuid reference of an agent to use.
-
+            ignore_agent_host: bool, optional, flag to ignore agent host
         Returns:
             str
         """
         if not self.settings:
             self.refresh_data()
-        if agent:
+        if agent and not ignore_agent_host:
             self._reset_agent(agent)
-
-        self._use_agent_host()
+        if not ignore_agent_host:
+            self._use_agent_host()
 
         params = get_streams_params(connection=self.artifacts_store)
         url = get_proxy_run_url(
@@ -1568,6 +1569,7 @@ class RunClient(ClientMixin):
         overwrite: bool = True,
         relative_to: Optional[str] = None,
         agent: Optional[str] = None,
+        ignore_agent_host: bool = False,
     ):
         """Uploads a full directory to the run's artifacts store path.
 
@@ -1581,6 +1583,7 @@ class RunClient(ClientMixin):
             relative_to: str, optional, if the path uploaded is not the current dir,
                  and you want to cancel the relative path.
             agent: str, optional, uuid reference of an agent to use.
+            ignore_agent_host: bool, optional, flag to ignore agent host
         Returns:
             str.
         """
@@ -1601,6 +1604,7 @@ class RunClient(ClientMixin):
             overwrite=overwrite,
             relative_to=relative_to,
             agent=agent,
+            ignore_agent_host=ignore_agent_host,
         )
 
     @client_handler(check_no_op=True, check_offline=True)
@@ -1611,6 +1615,7 @@ class RunClient(ClientMixin):
         overwrite: bool = True,
         relative_to: Optional[str] = None,
         agent: Optional[str] = None,
+        ignore_agent_host: bool = False,
     ):
         """Uploads multiple artifacts to the run's artifacts store path.
 
@@ -1621,6 +1626,7 @@ class RunClient(ClientMixin):
             relative_to: str, optional, if the path uploaded is not the current dir,
                  and you want to cancel the relative path.
             agent: str, optional, uuid reference of an agent to use.
+            ignore_agent_host: bool, optional, flag to ignore agent host
         Returns:
             str.
         """
@@ -1630,10 +1636,10 @@ class RunClient(ClientMixin):
 
         if not self.settings:
             self.refresh_data()
-        if agent:
+        if agent and not ignore_agent_host:
             self._reset_agent(agent)
-
-        self._use_agent_host()
+        if not ignore_agent_host:
+            self._use_agent_host()
 
         params = get_streams_params(connection=self.artifacts_store)
         url = get_proxy_run_url(
@@ -3019,6 +3025,7 @@ class RunClient(ClientMixin):
         upload_artifacts: bool = True,
         clean: bool = False,
         agent: Optional[str] = None,
+        ignore_agent_host: bool = False,
     ):
         """Syncs an offline run to Polyaxon's API and artifacts store.
 
@@ -3028,6 +3035,7 @@ class RunClient(ClientMixin):
             upload_artifacts: bool, optional, flag to trigger artifacts upload.
             clean: bool, optional, flag to clean local path after pushing the run.
             agent: str, optional, uuid reference of an agent to use.
+            ignore_agent_host: bool, optional, flag to ignore agent host
         """
         # We ensure that the is_offline is False
         is_offline = self._is_offline
@@ -3061,6 +3069,7 @@ class RunClient(ClientMixin):
                 overwrite=True,
                 relative_to=path,
                 agent=agent,
+                ignore_agent_host=ignore_agent_host,
             )
             logger.info(f"Offline artifacts for run {self.run_data.uuid} uploaded")
 

polyaxon/_sidecar/container/__init__.py

@@ -126,10 +126,10 @@ async def start_sidecar(
             if update_last_check:
                 state["last_artifacts_check"] = now()
 
-    while is_running and retry <= 3:
+    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:

polyaxon/_sidecar/container/monitors/logs.py

@@ -4,7 +4,7 @@ from typing import Optional
 
 import aiofiles
 
-from clipped.utils.paths import check_or_create_path
+from clipped.utils.paths import check_or_create_path, set_permissions
 from kubernetes_asyncio.client.models import V1Pod
 
 from polyaxon._contexts import paths as ctx_paths
@@ -34,8 +34,8 @@ async def sync_logs(
 
     path_from = "{}/{}.jsonl".format(path_from, pod.metadata.name)
     check_or_create_path(path_from, is_dir=False)
-    async with aiofiles.open(path_from, "a") as filepath:
+    async with aiofiles.open(path_from, "a") as outfile:
         _logs = V1Logs.construct(logs=logs)
-        await filepath.write(_logs.get_jsonl_events())
-
+        await outfile.write(_logs.get_jsonl_events())
+    set_permissions(path_from)
     return last_time

polyaxon/_sidecar/container/monitors/spec.py

@@ -1,7 +1,7 @@
 import aiofiles
 
 from clipped.utils.json import orjson_dumps
-from clipped.utils.paths import check_or_create_path
+from clipped.utils.paths import check_or_create_path, set_permissions
 
 from polyaxon._contexts import paths as ctx_paths
 from polyaxon._flow import V1RunKind
@@ -20,5 +20,6 @@ async def sync_spec(
     path_from = ctx_paths.CONTEXT_MOUNT_ARTIFACTS_FORMAT.format(run_uuid)
     path_from = "{}/outputs/spec.json".format(path_from)
     check_or_create_path(path_from, is_dir=False)
-    async with aiofiles.open(path_from, "w") as filepath:
-        await filepath.write(orjson_dumps(op_spec))
+    async with aiofiles.open(path_from, "w") as outfile:
+        await outfile.write(orjson_dumps(op_spec))
+    set_permissions(path_from)

polyaxon/pkg.py

@@ -1,5 +1,5 @@
 NAME = "polyaxon"
-VERSION = "2.7.0-rc0"
+VERSION = "2.8.0-rc0"
 SCHEMA_VERSION = 1.1
 DESC = "Command Line Interface (CLI) and client to interact with Polyaxon API."
 URL = "https://github.com/polyaxon/polyaxon"

{polyaxon-2.7.0rc0.dist-info → polyaxon-2.8.0rc0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyaxon
-Version: 2.7.0rc0
+Version: 2.8.0rc0
 Summary: Command Line Interface (CLI) and client to interact with Polyaxon API.
 Home-page: https://github.com/polyaxon/polyaxon
 Author: Polyaxon, Inc.
@@ -46,7 +46,7 @@ Requires-Dist: psutil>=5.4.7
 Requires-Dist: requests>=2.20.1
 Requires-Dist: requests-toolbelt>=0.8.0
 Requires-Dist: rich>=12.0.0
-Requires-Dist: sentry-sdk<2.6,>=1.2.0
+Requires-Dist: sentry-sdk>=1.2.0
 Requires-Dist: urllib3>=1.25.6
 Requires-Dist: certifi>=2022.12.7
 Requires-Dist: pydantic>=1.10.2
@@ -60,9 +60,9 @@ Provides-Extra: docker
 Requires-Dist: docker; extra == "docker"
 Provides-Extra: fs
 Requires-Dist: adlfs==2024.12.0; extra == "fs"
-Requires-Dist: fsspec==2025.2.0; extra == "fs"
-Requires-Dist: gcsfs==2025.2.0; extra == "fs"
-Requires-Dist: s3fs==2025.2.0; extra == "fs"
+Requires-Dist: fsspec==2025.3.2; extra == "fs"
+Requires-Dist: gcsfs==2025.3.2; extra == "fs"
+Requires-Dist: s3fs==2025.3.2; extra == "fs"
 Provides-Extra: fsspec
 Requires-Dist: fsspec; extra == "fsspec"
 Provides-Extra: gcs
@@ -93,7 +93,7 @@ Requires-Dist: anyio; extra == "sidecar"
 [![Slack](https://img.shields.io/badge/Slack-1.5k%20members-blue.svg?style=flat&logo=slack&longCache=true)](https://polyaxon.com/slack/)
 
 [![Docs](https://img.shields.io/badge/docs-stable-brightgreen.svg?style=flat&longCache=true)](https://polyaxon.com/docs/)
-[![Release](https://img.shields.io/badge/release-v2.7.0-brightgreen.svg?longCache=true)](https://polyaxon.com/docs/releases/2-7/)
+[![Release](https://img.shields.io/badge/release-v2.8.0-brightgreen.svg?longCache=true)](https://polyaxon.com/docs/releases/2-8/)
 [![GitHub](https://img.shields.io/badge/issue_tracker-github-blue?style=flat&logo=github&longCache=true)](https://github.com/polyaxon/polyaxon/issues)
 [![GitHub](https://img.shields.io/badge/roadmap-github-blue?style=flat&logo=github&longCache=true)](https://github.com/orgs/polyaxon/projects/5)
 

{polyaxon-2.7.0rc0.dist-info → polyaxon-2.8.0rc0.dist-info}/RECORD

@@ -9,7 +9,7 @@ polyaxon/exceptions.py,sha256=ujvG9p1Pn2KHYbHqB3-faadW46dEuULUQXNtfkd2zk8,10236
 polyaxon/fs.py,sha256=RS8XmVrrfXfIJXN6cTCCRRYwesCLHVVfC01Vi56lecs,246
 polyaxon/k8s.py,sha256=nI5oPCSlqU4aaeVShM6SlYS9eqYiYUL4GDXIZ4bnq-I,1051
 polyaxon/logger.py,sha256=gdZQms37Pe5G2j-Ear5jbSAJeGgX6fnvg7oE8_9MSlc,2309
-polyaxon/pkg.py,sha256=S64Hj0dQslx8Bo0GHJPf_2Qs8l1sNW_0g4ZwFG7uZ54,266
+polyaxon/pkg.py,sha256=rNanmnwwUnkdmulthVq7Ty7ue-wcVM_6hwXEcytqp8s,266
 polyaxon/polyaxonfile.py,sha256=xHmHT_cHomfuAQm82Jhnp71YNN5mQ-Lod7EbonjY4b4,429
 polyaxon/schemas.py,sha256=-CykY3emoAUCs_zRNbjwkuMkqbaEDjfKsZC86rI8154,5870
 polyaxon/settings.py,sha256=Pxx1-T2oeJ5XmvGFN0YgnVzum_9FyTPaQtl68aQvYc4,4116
@@ -33,7 +33,7 @@ polyaxon/_cli/dashboard.py,sha256=Dmx6lXqVmUQ60eOtEpxWCvGzKEjibxXVmSHDQUkRn8M,16
 polyaxon/_cli/errors.py,sha256=BYs7-I0CLLNYma-I0eJXB1EHs8ZTs_nWGRKGJz6CzTI,1031
 polyaxon/_cli/init.py,sha256=lZhkIbaHGv4FdMypdmXhYXRrH5GpM-qcDEks4IsmKIM,7714
 polyaxon/_cli/models.py,sha256=71oxGrIe4nQORnnfAiNi6T7A0DBnEe3V6whBABBGL3c,18066
-polyaxon/_cli/operations.py,sha256=ywF4qBMrVpVdzFs1qwUTXX0XZ8-VMnAsyzklpczpLaU,76272
+polyaxon/_cli/operations.py,sha256=4J9smfD9vN_4cLOhDvTsdQuJ0M7JLlT6aXsF1ieuqz4,76867
 polyaxon/_cli/options.py,sha256=-jeMZsdfg0JOV_QzVDf1hAhqK55NI0dkC_x4MZZWty8,1927
 polyaxon/_cli/port_forward.py,sha256=Lshpcrv7-4tXcriHmppiFW_3QZ7ZosDtUbJDIvdddSA,2733
 polyaxon/_cli/project_versions.py,sha256=fbgE3tRShrgH8TAA6ETj78J38HG2-BcVoAWbxYS9d5s,20507
@@ -57,7 +57,7 @@ polyaxon/_client/impersonate.py,sha256=4jRoJoX8nkwvVc3zAYqVqPoysNT1kZoPRbwyTpyHG
 polyaxon/_client/init.py,sha256=QECGjzuBTFaagndubzd9U1COpc5NGe-E0aVkSRetcIs,418
 polyaxon/_client/mixin.py,sha256=-tPm8RBej0UeG9HAKH3qN2fPX8PBtuXAHY1OpwLIDu4,1017
 polyaxon/_client/project.py,sha256=-vHLae0fMYqOjMscRm0I25Ixj-hS6o9i6XQm5PIxyiA,65520
-polyaxon/_client/run.py,sha256=Zj0sXR2c_80fqxm-jOq4IudgtjFiQ6MoWkKb1pfpr8o,120362
+polyaxon/_client/run.py,sha256=l5rlBzw7EE3nf8wRDCWS0-fCkXserxrrMQ1Q4ewOZAQ,121045
 polyaxon/_client/store.py,sha256=-Y33ypkijGQnhHTQ_vCTqLlpk0wRqoaP-ntJhdUtv7E,11311
 polyaxon/_client/decorators/__init__.py,sha256=e5CBijciLP-Ic-YkaL4tFhUdr--uod_TexvxAJamGZQ,186
 polyaxon/_client/decorators/client_call_handler.py,sha256=dgdidWsL9e8AgJmOfxz4urkI3bBgX64whWI1WHeAJDU,4587
@@ -563,12 +563,12 @@ polyaxon/_services/values.py,sha256=Uyj-kB5omSm3SbwpjqeuboSSOK5dyovoh-RszHPdiQ8,
 polyaxon/_sidecar/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 polyaxon/_sidecar/ignore.py,sha256=VSMo0kaTD7K2zTyVJ1u84mZyR_07YdICptbKsxHKo9c,26
 polyaxon/_sidecar/processor.py,sha256=XIIxl_9W2wme0xOq-4lBnZeThJrGj0dlaC4_mhSKgVI,3235
-polyaxon/_sidecar/container/__init__.py,sha256=jVnscR63EyOfRg9Y6H3TYDx_v8C7jnrYfk8DJxVFmSI,5823
+polyaxon/_sidecar/container/__init__.py,sha256=rXt25xnx2VRKATOZQxqhSlkXjR5-Qmh3_koYKOKR9pc,5809
 polyaxon/_sidecar/container/intervals.py,sha256=-tckK4E_Ur0feVsLBhUPUu6brx-JaU6RDFQBW-TAo6I,304
 polyaxon/_sidecar/container/monitors/__init__.py,sha256=x12AMMGTi4cK_LaDvEK0gd3jZ3R57iPnp5ICehCwOl8,202
 polyaxon/_sidecar/container/monitors/artifacts.py,sha256=BlXMOiKJFhHcRpJ9T7izh-GtcXoKfulZmzfq7aBgdOY,2746
-polyaxon/_sidecar/container/monitors/logs.py,sha256=xionA2GZCfcbtBVYpzTBLnkfWlFsG1wT1y1Y9LPc0Ik,1197
-polyaxon/_sidecar/container/monitors/spec.py,sha256=1JCkz0gH8j6C9WNlG6c4_0PiwfghFXc7bv3VeE13Mhw,836
+polyaxon/_sidecar/container/monitors/logs.py,sha256=0ENRmHsuMdWBBBwvmZHeeyH4LyoISIXJXrzYFxuO9ms,1242
+polyaxon/_sidecar/container/monitors/spec.py,sha256=gNmVnQPd7JacfUwRw-qeurdHzg4TDRUicNPNrhrn8oo,882
 polyaxon/_utils/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 polyaxon/_utils/cache.py,sha256=hsnADZUnnYKQ2jrzuQjg3z9_4YGznL-pof9wH-cObR8,1717
 polyaxon/_utils/cli_constants.py,sha256=y2siZNMaTZ7j_BlSrQv0BXpYHyPHzZxaTyPINyJSgCM,816
@@ -611,9 +611,9 @@ polyaxon/tuners/hyperopt.py,sha256=zd6MblMGkooqLGDFJVo5kClqYnBoMwGj-opqqj8FDzQ,7
 polyaxon/tuners/mapping.py,sha256=pOdHCiwEufTk-QT7pNyjBjAEWNTM-lMC17WNTCk7C24,69
 polyaxon/tuners/random_search.py,sha256=6VEekM3N9h6E1lbpVTTUGKFPJlGMY2u-GkG615_nQcI,80
 polyaxon_sdk/__init__.py,sha256=HWvFdGWESyVG3f26K_szewiG-McMOHFkXKTfZcBlHsM,92
-polyaxon-2.7.0rc0.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
-polyaxon-2.7.0rc0.dist-info/METADATA,sha256=Bs2dLEuGQV8NUVE526ZDFobkTM8tTcXbPKrbWPvciUw,11663
-polyaxon-2.7.0rc0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-polyaxon-2.7.0rc0.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
-polyaxon-2.7.0rc0.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
-polyaxon-2.7.0rc0.dist-info/RECORD,,
+polyaxon-2.8.0rc0.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
+polyaxon-2.8.0rc0.dist-info/METADATA,sha256=hDgaWEJFrHtFokFyfhgFklLnFlKqSKNLgZQhs8lYELk,11658
+polyaxon-2.8.0rc0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+polyaxon-2.8.0rc0.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
+polyaxon-2.8.0rc0.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
+polyaxon-2.8.0rc0.dist-info/RECORD,,