polyaxon 2.2.0.post1__py3-none-any.whl → 2.2.0rc1__py3-none-any.whl

polyaxon/_cli/operations.py

@@ -51,7 +51,7 @@ from polyaxon.api import (
     REWRITE_SERVICES_V1,
     SERVICES_V1,
 )
-from polyaxon.client import RunClient, V1Run, V1RunSettings, get_run_logs
+from polyaxon.client import RunClient, V1Run, get_run_logs
 from polyaxon.exceptions import (
     ApiException,
     PolyaxonClientException,
@@ -86,14 +86,6 @@ DEFAULT_EXCLUDE = [
 ]
 
 
-def get_op_agent_host(runSettings: V1RunSettings):
-    host_kwargs = {}
-    if runSettings and runSettings.agent and runSettings.agent.url:
-        host_kwargs["host"] = runSettings.agent.url
-
-    return host_kwargs
-
-
 def handle_run_statuses(status, conditions, table):
     if not conditions:
         return False
@@ -2097,12 +2089,10 @@ def service(ctx, project, uid, yes, external, url):
 
     wait_for_running_condition(client)
 
-    host_kwargs = get_op_agent_host(client.settings)
     run_url = get_dashboard_url(
         subpath="{}/runs/{}/service".format(
             get_project_subpath_url(owner, team, project_name), run_uuid
-        ),
-        **host_kwargs,
+        )
     )
 
     namespace = client.run_data.settings.namespace
@@ -2126,9 +2116,9 @@ def service(ctx, project, uid, yes, external, url):
     if port:
         service_subpath = "{}{}/".format(service_subpath, port)
 
-    host_kwargs = get_op_agent_host(client.settings)
     external_run_url = get_dashboard_url(
-        base=service_endpoint, subpath=service_subpath, **host_kwargs
+        base=service_endpoint,
+        subpath=service_subpath,
     )
 
     if url:

polyaxon/_cli/run.py

@@ -14,7 +14,6 @@ from polyaxon._cli.dashboard import get_dashboard_url, get_project_subpath_url
 from polyaxon._cli.errors import handle_cli_error
 from polyaxon._cli.operations import approve
 from polyaxon._cli.operations import execute as run_execute
-from polyaxon._cli.operations import get_op_agent_host
 from polyaxon._cli.operations import logs as run_logs
 from polyaxon._cli.operations import shell as run_shell
 from polyaxon._cli.operations import statuses
@@ -99,7 +98,13 @@ def _run(
                 meta_info=meta_info,
                 pending=pending,
             )
-            host_kwargs = get_op_agent_host(response.settings)
+            host_kwargs = {}
+            if (
+                response.settings
+                and response.settings.agent
+                and response.settings.agent.host
+            ):
+                host_kwargs["host"] = response.settings.agent.host
             run_url = get_dashboard_url(
                 subpath="{}/runs/{}".format(project_url, response.uuid), **host_kwargs
             )

polyaxon/_client/run.py

@@ -2136,9 +2136,9 @@ class RunClient(ClientMixin):
         abspath = filepath if is_abs() else os.path.abspath(filepath)
 
         for_patterns = []
-        if getattr(self, "_artifacts_path", None):
+        if getattr(self, "_artifacts_path"):
             for_patterns.append(getattr(self, "_artifacts_path"))
-        if getattr(self, "_store_path", None):
+        if getattr(self, "_store_path"):
             for_patterns.append(getattr(self, "_store_path"))
         context_root = (
             ctx_paths.CONTEXT_OFFLINE_ROOT

polyaxon/_compiler/contexts/ray_job.py

@@ -33,9 +33,7 @@ class RayJobContextsManager(BaseContextsManager):
                 connection_by_names=connection_by_names,
             )
 
-        data = {
+        return {
             "head": _get_replica(job.head),
+            "workers": {wn: _get_replica(job.workers[wn]) for wn in job.workers},
         }
-        if job.workers:
-            data["workers"] = {wn: _get_replica(job.workers[wn]) for wn in job.workers}
-        return data

polyaxon/_k8s/converter/converters/ray_job.py

@@ -58,9 +58,7 @@ class RayJobConverter(RayJobMixin, BaseConverter):
             config=compiled_operation.plugins, auth=default_auth
         )
         head = _get_replica(job.head)
-        workers = None
-        if job.workers:
-            workers = {n: _get_replica(w) for n, w in job.workers.items()}
+        workers = {n: _get_replica(w) for n, w in job.workers.items()}
         labels = self.get_labels(version=pkg.VERSION, labels={})
 
         return get_ray_job_custom_resource(
@@ -70,7 +68,7 @@ class RayJobConverter(RayJobMixin, BaseConverter):
             workers=workers,
             entrypoint=job.entrypoint,
             metadata=job.metadata,
-            runtime_env=orjson_dumps(job.runtime_env),
+            runtime_env=encode(orjson_dumps(job.runtime_env)),
             ray_version=job.ray_version,
             termination=compiled_operation.termination,
             collect_logs=plugins.collect_logs,

polyaxon/_k8s/custom_resources/setter.py

@@ -46,7 +46,7 @@ def set_notify(custom_object: Dict, notifications: List[V1Notification]) -> Dict
 def set_clean_pod_policy(template_spec: Dict, clean_pod_policy: str) -> Dict:
     if not clean_pod_policy:
         # Sets default clean pod policy
-        clean_pod_policy = "None"
+        clean_pod_policy = "All"
 
     template_spec["cleanPodPolicy"] = clean_pod_policy.capitalize()
     return template_spec

polyaxon/_sdk/schemas/v1_user.py

@@ -2,11 +2,12 @@ from typing import Optional
 
 from clipped.compact.pydantic import StrictStr
 from clipped.config.schema import BaseAllowSchemaModel
+from clipped.types.email import EmailStr
 
 
 class V1User(BaseAllowSchemaModel):
     username: Optional[StrictStr]
-    email: Optional[StrictStr]
+    email: Optional[EmailStr]
     name: Optional[StrictStr]
     kind: Optional[StrictStr]
     theme: Optional[int]

polyaxon/pkg.py

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

{polyaxon-2.2.0.post1.dist-info → polyaxon-2.2.0rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyaxon
-Version: 2.2.0.post1
+Version: 2.2.0rc1
 Summary: Command Line Interface (CLI) and client to interact with Polyaxon API.
 Home-page: https://github.com/polyaxon/polyaxon
 Author: Polyaxon, Inc.
@@ -36,7 +36,7 @@ License-File: LICENSE
 Requires-Dist: click <9.0.0,>=7.1.1
 Requires-Dist: clipped ==0.7.*
 Requires-Dist: vents ==0.4.*
-Requires-Dist: Jinja2 <3.1.5,>=2.10.3
+Requires-Dist: Jinja2 <3.1.3,>=2.10.3
 Requires-Dist: kubernetes >=10.0.1
 Requires-Dist: python-dateutil >=2.7.3
 Requires-Dist: pytz >=2019.2
@@ -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.40,>=1.2.0
 Requires-Dist: urllib3 >=1.25.6
 Requires-Dist: certifi >=2022.12.7
 Requires-Dist: pydantic >=1.10.2
@@ -59,10 +59,10 @@ Requires-Dist: moto ==2.0.5 ; extra == 'dev'
 Provides-Extra: docker
 Requires-Dist: docker ; extra == 'docker'
 Provides-Extra: fs
-Requires-Dist: adlfs ==2024.4.1 ; extra == 'fs'
-Requires-Dist: fsspec ==2024.6.0 ; extra == 'fs'
-Requires-Dist: gcsfs ==2024.6.0 ; extra == 'fs'
-Requires-Dist: s3fs ==2024.6.0 ; extra == 'fs'
+Requires-Dist: adlfs ==2024.4.0 ; extra == 'fs'
+Requires-Dist: fsspec ==2024.3.1 ; extra == 'fs'
+Requires-Dist: gcsfs ==2024.3.1 ; extra == 'fs'
+Requires-Dist: s3fs ==2024.3.1 ; extra == 'fs'
 Provides-Extra: fsspec
 Requires-Dist: fsspec ; extra == 'fsspec'
 Provides-Extra: gcs

{polyaxon-2.2.0.post1.dist-info → polyaxon-2.2.0rc1.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=oLf0FYbl74rgRk0sKKPmXHPQ_wIDqxfsaNmzBO4vdN0,268
+polyaxon/pkg.py,sha256=nw98is-vNGTWgbrOB6odUDJsBip0OfnZPvnlgW43Ipg,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,12 +33,12 @@ polyaxon/_cli/dashboard.py,sha256=zRrOJMfowM7YQnYXhz_guKsTi-jkoCuVl_g0UTJ37qE,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=iQTVHkVvUxFCRlaKBLP0xVxuxcb9KJHgTrtNe4Rf8LM,75624
+polyaxon/_cli/operations.py,sha256=AqcO4EtMA8sa7sdBhdG2YP3z_G-CzTBWFUobNirNvUc,75255
 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
 polyaxon/_cli/projects.py,sha256=RVuTbKtNP_bvw3kRFKaIQBWq0xUwF0BUHK19T87VG5Q,12069
-polyaxon/_cli/run.py,sha256=3KM_E0iLw8WUe36zeXuwMA5urzGljo6yf4A-DL1uz6k,17737
+polyaxon/_cli/run.py,sha256=j3n0C7BvbgxboaJjFbJ4ty4OcAckoRzkbWepp5--DOo,17874
 polyaxon/_cli/session.py,sha256=5Plolpxwkv3wiup0pRYBpBPD64dvb4GzeRPz_nuIGRo,6033
 polyaxon/_cli/utils.py,sha256=jRD1npuD89-0SxtzAx5SE3i4Lvhi8vnfYkmyOMbDxEY,1319
 polyaxon/_cli/version.py,sha256=bC0T8S6Momr7EvHGV9Swn868x9ZHD7oL8ahQv57yInk,4536
@@ -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=EyuuEL7-oabgv_AlX0osKT8flY-3eLMGOINgBk93xX4,118810
+polyaxon/_client/run.py,sha256=7aUowwMB_yDCFYrlBdYuh-LCKcsg7IUbSdaLusoYBX8,118798
 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
@@ -80,7 +80,7 @@ polyaxon/_compiler/contexts/base.py,sha256=4EVQQlLgxvFORZ27ouBhwcJKhl3vMGXyOVP9H
 polyaxon/_compiler/contexts/contexts.py,sha256=tJy8eL9UrUhjisJ3MdYNmMVERSK_m_fKfgPTQgC-0y4,6535
 polyaxon/_compiler/contexts/dask_job.py,sha256=GxhLcZtCDWSDfc45UenmKsJ6Fhr0f_GeneaAzLHtncg,1265
 polyaxon/_compiler/contexts/job.py,sha256=1lkjfMO2eHndmmtxC551K6n6RE5XasqkWBfwNpYy_MA,759
-polyaxon/_compiler/contexts/ray_job.py,sha256=6CFFuX63HhyU0z--GoE3Tv2vENBB1QKGCwuFmEwC2PY,1292
+polyaxon/_compiler/contexts/ray_job.py,sha256=rdf__86SFoUEYf5T3vUTZXGLL-QI10enYwyngXhEJ7Q,1242
 polyaxon/_compiler/contexts/service.py,sha256=d4zKB608906CXRZeUeOLgOdro5CEFe9byHWrGyy1ZFw,2150
 polyaxon/_compiler/contexts/kubeflow/__init__.py,sha256=FnOOO9l-eunvun5kGcTfQ9uO8gLhuHIFuh2_gCHxaLA,488
 polyaxon/_compiler/contexts/kubeflow/mpi_job.py,sha256=UMf7yJPRWEB9MVYS4Eoa7V6IZEQasU5VU71WdNZQ-2k,1214
@@ -310,7 +310,7 @@ polyaxon/_k8s/converter/converters/__init__.py,sha256=jyw6PnVQcG-lfyNbfxomFhh1aa
 polyaxon/_k8s/converter/converters/dask_job.py,sha256=DNlQczpN29AMJzsAy3q6SlpIH9h-kfcp3UMvfRLGJm8,2784
 polyaxon/_k8s/converter/converters/helpers.py,sha256=LEfPISHptnC_1W-B5lY3D834U0qAr6wsX4x1EafWlk4,661
 polyaxon/_k8s/converter/converters/job.py,sha256=luYWU9-qkuT5usk42zmd2JkX8thwWHNcqrmngtbHPJ8,2287
-polyaxon/_k8s/converter/converters/ray_job.py,sha256=d57kxNxt3VqN5Bu54NKFqH-HHyv0F2Ps74qJTNVNxSk,3264
+polyaxon/_k8s/converter/converters/ray_job.py,sha256=gmoHJ0lyeXU1QMpCsptU2ur_T16zxuLAhh3G8KCVYQ0,3221
 polyaxon/_k8s/converter/converters/service.py,sha256=F2tW2bTlpaX9J47I8nE6up8xwUbdg2dQ_oXqsJdcdu8,2465
 polyaxon/_k8s/converter/converters/kubeflow/__init__.py,sha256=ZXdpmleFm0ALmJYPlU9JG6ZpsBW6bq6DvF1d9IVoW68,498
 polyaxon/_k8s/converter/converters/kubeflow/mpi_job.py,sha256=UyLqLBCmjGIWlj6ZiDJqMByzeVWTKp1wBtu4oogfL3w,2731
@@ -329,7 +329,7 @@ polyaxon/_k8s/custom_resources/job.py,sha256=hTETu_oU90y9CujwflodopZ_HMqi3ajMPsy
 polyaxon/_k8s/custom_resources/operation.py,sha256=qrtxZ6i19N5tZIYKqGRoqCM3dvQg5lSVC1mZMo0Kx3w,625
 polyaxon/_k8s/custom_resources/ray_job.py,sha256=Ytvsiaev4xSL5Y3SN8vcI15DZbStkOZJq11GaZa-T1g,6210
 polyaxon/_k8s/custom_resources/service.py,sha256=M_mptT7JR6jmuctfgo87ov7KXRM4_jggJ5nc3zXL7hA,2370
-polyaxon/_k8s/custom_resources/setter.py,sha256=WCbEf7DaijK2U-FNAPAVI7mEefA7rb-KhHYnWPRX6aU,1846
+polyaxon/_k8s/custom_resources/setter.py,sha256=YgTV0vnTEHMibISHKkt3IRA6cozW-6ynXw8fGfcALpo,1845
 polyaxon/_k8s/custom_resources/kubeflow/__init__.py,sha256=oCDo38kAVl7L1ehQTfWX04fZHRTa-V5qYKbef9-jG7k,556
 polyaxon/_k8s/custom_resources/kubeflow/common.py,sha256=_OrM906v50aTziJcGUOQqzr5XYBuTcNeErgAau2ukng,1041
 polyaxon/_k8s/custom_resources/kubeflow/mpi_job.py,sha256=GwBfAb4LVYvDdHdzBUW2QgCq-_ypEhKyPrkInkaXyZo,2565
@@ -548,7 +548,7 @@ polyaxon/_sdk/schemas/v1_team_member.py,sha256=Sfj3-GH6iwf5auTbj9u3Uy6NGgYY5HP9n
 polyaxon/_sdk/schemas/v1_team_settings.py,sha256=bfWhOItuTCOnQQ15jkaYQWH4kwBKJI8QkwKj4vJ9xnw,301
 polyaxon/_sdk/schemas/v1_token.py,sha256=g2g2bfPtgBcZskfou-RO5sfDP9FqEoXYBgxoOkIhjC8,604
 polyaxon/_sdk/schemas/v1_trial_start.py,sha256=h1Ut9R529Vsfug8HRtDEdjWlmTEWAPOsvbxavJAb4hQ,418
-polyaxon/_sdk/schemas/v1_user.py,sha256=HmbtjJavgA7GkSQIPN2xqFG2hNFEj5zjf1xg0m8dpuk,357
+polyaxon/_sdk/schemas/v1_user.py,sha256=BOb30Zw0VN-VfW4hNdZ_u0FygBKE-YgSWm7Qy6qxC0M,397
 polyaxon/_sdk/schemas/v1_user_access.py,sha256=TwFqN1RFBhvE8JDopjikChQV0OQewuVLYoNR_J0fDaE,303
 polyaxon/_sdk/schemas/v1_user_email.py,sha256=BpMbTfccBdFr_s5i2KO_d_6MXRz_lcY94WTHdmZ1AdA,198
 polyaxon/_sdk/schemas/v1_user_singup.py,sha256=eqG5Uic8CBHtnDRh7LWf-hO6K6M2pFPXQhYOb6-K1pk,392
@@ -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.2.0.post1.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
-polyaxon-2.2.0.post1.dist-info/METADATA,sha256=BURB4v7eJBY_R0T1NDcn9WsKg-lef-4BCY2eIqPXMF8,11714
-polyaxon-2.2.0.post1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-polyaxon-2.2.0.post1.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
-polyaxon-2.2.0.post1.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
-polyaxon-2.2.0.post1.dist-info/RECORD,,
+polyaxon-2.2.0rc1.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
+polyaxon-2.2.0rc1.dist-info/METADATA,sha256=ISDUCO1LOLCX0kzJaU4eFgD4xCrUOG2HJ6SpKE5xZNE,11712
+polyaxon-2.2.0rc1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+polyaxon-2.2.0rc1.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
+polyaxon-2.2.0rc1.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
+polyaxon-2.2.0rc1.dist-info/RECORD,,