polyaxon 2.2.0rc1__tar.gz → 2.3.0rc0__tar.gz

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyaxon
-Version: 2.2.0rc1
+Version: 2.3.0rc0
 Summary: Command Line Interface (CLI) and client to interact with Polyaxon API.
 Home-page: https://github.com/polyaxon/polyaxon
 Author: Polyaxon, Inc.
@@ -51,7 +51,7 @@ Provides-Extra: sandbox
 [![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.2.0-brightgreen.svg?longCache=true)](https://polyaxon.com/docs/releases/2-1/)
+[![Release](https://img.shields.io/badge/release-v2.3.0-brightgreen.svg?longCache=true)](https://polyaxon.com/docs/releases/2-1/)
 [![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.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_cli/operations.py

@@ -51,7 +51,7 @@ from polyaxon.api import (
     REWRITE_SERVICES_V1,
     SERVICES_V1,
 )
-from polyaxon.client import RunClient, V1Run, get_run_logs
+from polyaxon.client import RunClient, V1Run, V1RunSettings, get_run_logs
 from polyaxon.exceptions import (
     ApiException,
     PolyaxonClientException,
@@ -86,6 +86,14 @@ 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
@@ -2089,10 +2097,12 @@ 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
@@ -2116,9 +2126,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,
+        base=service_endpoint, subpath=service_subpath, **host_kwargs
     )
 
     if url:

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_cli/run.py

@@ -14,6 +14,7 @@ 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
@@ -98,13 +99,7 @@ def _run(
                 meta_info=meta_info,
                 pending=pending,
             )
-            host_kwargs = {}
-            if (
-                response.settings
-                and response.settings.agent
-                and response.settings.agent.host
-            ):
-                host_kwargs["host"] = response.settings.agent.host
+            host_kwargs = get_op_agent_host(response.settings)
             run_url = get_dashboard_url(
                 subpath="{}/runs/{}".format(project_url, response.uuid), **host_kwargs
             )

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/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"):
+        if getattr(self, "_artifacts_path", None):
             for_patterns.append(getattr(self, "_artifacts_path"))
-        if getattr(self, "_store_path"):
+        if getattr(self, "_store_path", None):
             for_patterns.append(getattr(self, "_store_path"))
         context_root = (
             ctx_paths.CONTEXT_OFFLINE_ROOT

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_compiler/contexts/ray_job.py

@@ -33,7 +33,9 @@ class RayJobContextsManager(BaseContextsManager):
                 connection_by_names=connection_by_names,
             )
 
-        return {
+        data = {
             "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-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_k8s/converter/base/base.py

@@ -2,6 +2,7 @@ import copy
 
 from typing import Dict, Iterable, List, Optional
 
+from clipped.utils.enums import get_enum_value
 from clipped.utils.sanitizers import sanitize_string_dict
 from clipped.utils.strings import slugify
 
@@ -85,7 +86,7 @@ class BaseConverter(
             "operation.polyaxon.com/name": self.run_name,
             "operation.polyaxon.com/owner": self.owner_name,
             "operation.polyaxon.com/project": self.project_name,
-            "operation.polyaxon.com/kind": self.K8S_ANNOTATIONS_KIND,
+            "operation.polyaxon.com/kind": get_enum_value(self.K8S_ANNOTATIONS_KIND),
         }
 
     def get_annotations(

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_k8s/converter/converters/ray_job.py

@@ -58,7 +58,9 @@ class RayJobConverter(RayJobMixin, BaseConverter):
             config=compiled_operation.plugins, auth=default_auth
         )
         head = _get_replica(job.head)
-        workers = {n: _get_replica(w) for n, w in job.workers.items()}
+        workers = None
+        if job.workers:
+            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(
@@ -68,7 +70,7 @@ class RayJobConverter(RayJobMixin, BaseConverter):
             workers=workers,
             entrypoint=job.entrypoint,
             metadata=job.metadata,
-            runtime_env=encode(orjson_dumps(job.runtime_env)),
+            runtime_env=orjson_dumps(job.runtime_env),
             ray_version=job.ray_version,
             termination=compiled_operation.termination,
             collect_logs=plugins.collect_logs,

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_k8s/custom_resources/dask_job.py

@@ -46,6 +46,9 @@ def get_dask_replicas_template(
     if liveness_probe and replica.main_container.liveness_probe is None:
         replica.main_container.liveness_probe = liveness_probe
 
+    labels = {**labels, **replica.labels}
+    annotations = {**annotations, **replica.annotations}
+
     metadata, pod_spec = get_pod_spec(
         namespace=namespace,
         main_container=replica.main_container,

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_k8s/custom_resources/kubeflow/common.py

@@ -16,6 +16,9 @@ def get_kf_replicas_template(
     if not replica:
         return
 
+    labels = {**labels, **replica.labels}
+    annotations = {**annotations, **replica.annotations}
+
     metadata, pod_spec = get_pod_spec(
         namespace=namespace,
         main_container=replica.main_container,

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_k8s/custom_resources/ray_job.py

@@ -26,6 +26,9 @@ def _get_ray_replicas_template(
     if not replica:
         return
 
+    labels = {**labels, **replica.labels}
+    annotations = {**annotations, **replica.annotations}
+
     metadata, pod_spec = get_pod_spec(
         namespace=namespace,
         main_container=replica.main_container,

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/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 = "All"
+        clean_pod_policy = "None"
 
     template_spec["cleanPodPolicy"] = clean_pod_policy.capitalize()
     return template_spec

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_pql/manager.py

@@ -13,7 +13,7 @@ class PQLManager:
     FIELDS_PROXY = {}
     FIELDS_TRANS = {}
     FIELDS_ORDERING = None
-    FIELDS_ORDERING_PROXY = None
+    FIELDS_ORDERING_PROXY = None  # Do not set a field on both field and proxy
     FIELDS_DEFAULT_ORDERING = None
     FIELDS_DISTINCT = None
     CHECK_ALIVE = True

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_sdk/api/runs_v1_api.py

@@ -3014,8 +3014,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         kind: Annotated[str, Field(..., description="The artifact kind")],
         names: Annotated[
@@ -3046,15 +3046,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_events(namespace, owner, project, kind, names, runs, orient, force, sample, connection, status, async_req=True)
+        >>> thread = api.get_multi_run_events(namespace, owner, entity, kind, names, runs, orient, force, sample, connection, status, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param kind: The artifact kind (required)
         :type kind: str
         :param names: Names query param.
@@ -3090,7 +3090,7 @@ class RunsV1Api(BaseApi):
         return self.get_multi_run_events_with_http_info(
             namespace,
             owner,
-            project,
+            entity,
             kind,
             names,
             runs,
@@ -3107,8 +3107,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         kind: Annotated[str, Field(..., description="The artifact kind")],
         names: Annotated[
@@ -3139,15 +3139,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_events_with_http_info(namespace, owner, project, kind, names, runs, orient, force, sample, connection, status, async_req=True)
+        >>> thread = api.get_multi_run_events_with_http_info(namespace, owner, entity, kind, names, runs, orient, force, sample, connection, status, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param kind: The artifact kind (required)
         :type kind: str
         :param names: Names query param.
@@ -3193,7 +3193,7 @@ class RunsV1Api(BaseApi):
         _all_params = [
             "namespace",
             "owner",
-            "project",
+            "entity",
             "kind",
             "names",
             "runs",
@@ -3233,8 +3233,8 @@ class RunsV1Api(BaseApi):
             _path_params["namespace"] = _params["namespace"]
         if _params["owner"]:
             _path_params["owner"] = _params["owner"]
-        if _params["project"]:
-            _path_params["project"] = _params["project"]
+        if _params["entity"]:
+            _path_params["entity"] = _params["entity"]
         if _params["kind"]:
             _path_params["kind"] = _params["kind"]
 
@@ -3281,7 +3281,7 @@ class RunsV1Api(BaseApi):
         }
 
         return self.api_client.call_api(
-            "/streams/v1/{namespace}/{owner}/{project}/runs/multi/events/{kind}",
+            "/streams/v1/{namespace}/{owner}/{entity}/runs/multi/events/{kind}",
             "GET",
             _path_params,
             _query_params,
@@ -3304,8 +3304,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         body: Annotated[Dict[str, Any], Field(..., description="Params/Metrics data")],
         **kwargs
@@ -3315,15 +3315,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_importance(namespace, owner, project, body, async_req=True)
+        >>> thread = api.get_multi_run_importance(namespace, owner, entity, body, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param body: Params/Metrics data (required)
         :type body: object
         :param async_req: Whether to execute the request asynchronously.
@@ -3343,7 +3343,7 @@ class RunsV1Api(BaseApi):
         """
         kwargs["_return_http_data_only"] = True
         return self.get_multi_run_importance_with_http_info(
-            namespace, owner, project, body, **kwargs
+            namespace, owner, entity, body, **kwargs
         )  # noqa: E501
 
     @validate_arguments
@@ -3351,8 +3351,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         body: Annotated[Dict[str, Any], Field(..., description="Params/Metrics data")],
         **kwargs
@@ -3362,15 +3362,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_importance_with_http_info(namespace, owner, project, body, async_req=True)
+        >>> thread = api.get_multi_run_importance_with_http_info(namespace, owner, entity, body, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param body: Params/Metrics data (required)
         :type body: object
         :param async_req: Whether to execute the request asynchronously.
@@ -3399,7 +3399,7 @@ class RunsV1Api(BaseApi):
 
         _params = locals()
 
-        _all_params = ["namespace", "owner", "project", "body"]
+        _all_params = ["namespace", "owner", "entity", "body"]
         _all_params.extend(
             [
                 "async_req",
@@ -3432,8 +3432,8 @@ class RunsV1Api(BaseApi):
         if _params["owner"]:
             _path_params["owner"] = _params["owner"]
 
-        if _params["project"]:
-            _path_params["project"] = _params["project"]
+        if _params["entity"]:
+            _path_params["entity"] = _params["entity"]
 
         # process the query parameters
         _query_params = []
@@ -3471,7 +3471,7 @@ class RunsV1Api(BaseApi):
         }
 
         return self.api_client.call_api(
-            "/streams/v1/{namespace}/{owner}/{project}/runs/multi/importance",
+            "/streams/v1/{namespace}/{owner}/{entity}/runs/multi/importance",
             "POST",
             _path_params,
             _query_params,

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/_sdk/schemas/v1_user.py

@@ -2,12 +2,11 @@ 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[EmailStr]
+    email: Optional[StrictStr]
     name: Optional[StrictStr]
     kind: Optional[StrictStr]
     theme: Optional[int]

{polyaxon-2.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon/pkg.py

@@ -1,5 +1,5 @@
 NAME = "polyaxon"
-VERSION = "2.2.0-rc1"
+VERSION = "2.3.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.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyaxon
-Version: 2.2.0rc1
+Version: 2.3.0rc0
 Summary: Command Line Interface (CLI) and client to interact with Polyaxon API.
 Home-page: https://github.com/polyaxon/polyaxon
 Author: Polyaxon, Inc.
@@ -51,7 +51,7 @@ Provides-Extra: sandbox
 [![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.2.0-brightgreen.svg?longCache=true)](https://polyaxon.com/docs/releases/2-1/)
+[![Release](https://img.shields.io/badge/release-v2.3.0-brightgreen.svg?longCache=true)](https://polyaxon.com/docs/releases/2-1/)
 [![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.2.0rc1 → polyaxon-2.3.0rc0}/polyaxon.egg-info/requires.txt

@@ -1,7 +1,7 @@
 click<9.0.0,>=7.1.1
 clipped==0.7.*
 vents==0.4.*
-Jinja2<3.1.3,>=2.10.3
+Jinja2<3.1.5,>=2.10.3
 kubernetes>=10.0.1
 python-dateutil>=2.7.3
 pytz>=2019.2
@@ -11,7 +11,7 @@ psutil>=5.4.7
 requests>=2.20.1
 requests-toolbelt>=0.8.0
 rich>=12.0.0
-sentry-sdk<1.40,>=1.2.0
+sentry-sdk<2.6,>=1.2.0
 urllib3>=1.25.6
 certifi>=2022.12.7
 pydantic>=1.10.2
@@ -28,10 +28,10 @@ moto==2.0.5
 docker
 
 [fs]
-adlfs==2024.4.0
-fsspec==2024.3.1
-gcsfs==2024.3.1
-s3fs==2024.3.1
+adlfs==2024.4.1
+fsspec==2024.6.1
+gcsfs==2024.6.1
+s3fs==2024.6.1
 
 [fsspec]
 fsspec