polyaxon 2.0.0rc50__tar.gz → 2.0.0rc51__tar.gz

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyaxon
-Version: 2.0.0rc50
+Version: 2.0.0rc51
 Summary: Command Line Interface (CLI) and client to interact with Polyaxon API.
 Home-page: https://github.com/polyaxon/polyaxon
 Author: Polyaxon, Inc.

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_client/run.py

@@ -58,6 +58,7 @@ from polyaxon._schemas.lifecycle import (
     V1Statuses,
 )
 from polyaxon._schemas.types import V1ArtifactsType
+from polyaxon._sdk.schemas import V1RunEdgeLineage, V1RunEdgesGraph
 from polyaxon._sdk.schemas.v1_operation_body import V1OperationBody
 from polyaxon._sdk.schemas.v1_project_version import V1ProjectVersion
 from polyaxon._sdk.schemas.v1_run import V1Run
@@ -2119,6 +2120,10 @@ class RunClient:
         if not self._has_meta_key("has_events"):
             self.log_meta(has_events=True)
 
+    def _log_has_traces(self):
+        if not self._has_meta_key("has_traces"):
+            self.log_meta(has_traces=True, has_events=True)
+
     def _log_has_metrics(self):
         data = {}
         if not self._has_meta_key("has_metrics"):
@@ -2646,6 +2651,21 @@ class RunClient:
             force=force,
         )
 
+    @client_handler(check_no_op=True, check_offline=True)
+    def set_run_edges_lineage(self, run_edges: List[V1RunEdgeLineage]):
+        """
+        Set run edges lineage.
+
+        Args:
+            run_edges: List[V1RunEdgeLineage], list of run edges.
+        """
+        return self.client.runs_v1.set_run_edges_lineage(
+            self.owner,
+            self.project,
+            self.run_uuid,
+            body=V1RunEdgesGraph(edges=run_edges),
+        )
+
     @client_handler(check_no_op=True, check_offline=True)
     def promote_to_artifact_version(
         self,
@@ -2669,7 +2689,7 @@ class RunClient:
             description: str, optional, the version description.
             tags: str or List[str], optional.
             content: str or dict, optional, content/metadata (JSON object) of the version.
-            connection: str, optional, a uuid reference to a connection.
+            connection: str, optional, uuid reference to a connection.
             artifacts: List[str], optional, list of artifacts to highlight(requires passing a run)
             force: bool, optional, to force push, i.e. update if exists.
 

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_compiler/contexts/contexts.py

@@ -55,7 +55,7 @@ def resolve_globals_contexts(
     schedule_at: Optional[datetime] = None,
     started_at: Optional[datetime] = None,
     finished_at: Optional[datetime] = None,
-    duration: Optional[int] = None,
+    duration: Optional[float] = None,
     plugins: Optional[V1Plugins] = None,
     artifacts_store: V1Connection = None,
     cloning_kind: V1CloningKind = None,
@@ -134,7 +134,7 @@ def resolve_contexts(
     schedule_at: Optional[datetime] = None,
     started_at: Optional[datetime] = None,
     finished_at: Optional[datetime] = None,
-    duration: Optional[int] = None,
+    duration: Optional[float] = None,
     cloning_kind: V1CloningKind = None,
     original_uuid: Optional[str] = None,
     is_independent: bool = True,

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_flow/run/enums.py

@@ -56,6 +56,7 @@ class V1RunEdgeKind(str, PEnum):
     RUN = "run"
     TB = "tb"
     BUILD = "build"
+    MANUAL = "manual"
 
 
 class V1RunPending(str, PEnum):

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_schemas/lifecycle.py

@@ -275,8 +275,8 @@ class LifeCycle:
             entity.started_at = now()
             # Update wait_time
             if entity.wait_time is None:
-                entity.wait_time = int(
-                    (entity.started_at - entity.created_at).total_seconds()
+                entity.wait_time = round(
+                    (entity.started_at - entity.created_at).total_seconds(), 2
                 )
             return True
 
@@ -290,8 +290,8 @@ class LifeCycle:
                 entity.started_at = entity.created_at
             # Update duration
             if entity.duration is None:
-                entity.duration = int(
-                    (entity.finished_at - entity.started_at).total_seconds()
+                entity.duration = round(
+                    (entity.finished_at - entity.started_at).total_seconds(), 2
                 )
             return True
         return False

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_sdk/api/organizations_v1_api.py

@@ -2862,7 +2862,9 @@ class OrganizationsV1Api(BaseApi):
         ] = None,
         organization_archived_deletion_interval: Annotated[
             Optional[StrictInt],
-            Field(description="Setting to configure default archived deletion interval.")
+            Field(
+                description="Setting to configure default archived deletion interval."
+            ),
         ] = None,
         **kwargs
     ) -> V1Organization:  # noqa: E501
@@ -2983,7 +2985,9 @@ class OrganizationsV1Api(BaseApi):
         ] = None,
         organization_archived_deletion_interval: Annotated[
             Optional[StrictInt],
-            Field(description="Setting to configure default archived deletion interval.")
+            Field(
+                description="Setting to configure default archived deletion interval."
+            ),
         ] = None,
         **kwargs
     ):  # noqa: E501
@@ -3142,7 +3146,9 @@ class OrganizationsV1Api(BaseApi):
                     _params["organization_is_cloud_viewable"],
                 )
             )
-        if _params.get("organization_archived_deletion_interval") is not None:  # noqa: E501
+        if (
+            _params.get("organization_archived_deletion_interval") is not None
+        ):  # noqa: E501
             _query_params.append(
                 (
                     "organization.archived_deletion_interval",

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_sdk/api/runs_v1_api.py

@@ -6,6 +6,7 @@ from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_argum
 
 from polyaxon._schemas.lifecycle import V1Status
 from polyaxon._sdk.base_api import BaseApi
+from polyaxon._sdk.schemas import V1RunEdgesGraph
 from polyaxon._sdk.schemas.v1_artifact_tree import V1ArtifactTree
 from polyaxon._sdk.schemas.v1_auth import V1Auth
 from polyaxon._sdk.schemas.v1_entities_tags import V1EntitiesTags
@@ -10573,6 +10574,185 @@ class RunsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
+    @validate_arguments
+    def set_run_edges_lineage(
+        self,
+        owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
+        project: Annotated[StrictStr, Field(..., description="Project")],
+        uuid: Annotated[StrictStr, Field(..., description="Run uuid")],
+        body: Annotated[V1RunEdgesGraph, Field(..., description="Run edges graph")],
+        **kwargs
+    ) -> None:  # noqa: E501
+        """Set run edges graph lineage  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.set_run_edges_lineage(owner, project, uuid, body, async_req=True)
+        >>> result = thread.get()
+
+        :param owner: Owner of the namespace (required)
+        :type owner: str
+        :param project: Project (required)
+        :type project: str
+        :param uuid: Run uuid (required)
+        :type uuid: str
+        :param body: Run edges graph (required)
+        :type body: V1RunEdgesGraph
+        :param async_req: Whether to execute the request asynchronously.
+        :type async_req: bool, optional
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :type _preload_content: bool, optional
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: Returns the result object.
+                 If the method is called asynchronously,
+                 returns the request thread.
+        :rtype: None
+        """
+        kwargs["_return_http_data_only"] = True
+        return self.set_run_edges_lineage_with_http_info(
+            owner, project, uuid, body, **kwargs
+        )  # noqa: E501
+
+    @validate_arguments
+    def set_run_edges_lineage_with_http_info(
+        self,
+        owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
+        project: Annotated[StrictStr, Field(..., description="Project")],
+        uuid: Annotated[StrictStr, Field(..., description="Run uuid")],
+        body: Annotated[V1RunEdgesGraph, Field(..., description="Run edges graph")],
+        **kwargs
+    ):  # noqa: E501
+        """Set run edges graph lineage  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.set_run_edges_lineage_with_http_info(owner, project, uuid, body, async_req=True)
+        >>> result = thread.get()
+
+        :param owner: Owner of the namespace (required)
+        :type owner: str
+        :param project: Project (required)
+        :type project: str
+        :param uuid: Run uuid (required)
+        :type uuid: str
+        :param body: Run edges graph (required)
+        :type body: V1RunEdgesGraph
+        :param async_req: Whether to execute the request asynchronously.
+        :type async_req: bool, optional
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :type _return_http_data_only: bool, optional
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :type _preload_content: bool, optional
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the authentication
+                              in the spec for a single request.
+        :type _request_auth: dict, optional
+        :type _content_type: string, optional: force content-type for the request
+        :return: Returns the result object.
+                 If the method is called asynchronously,
+                 returns the request thread.
+        :rtype: None
+        """
+
+        _params = locals()
+
+        _all_params = ["owner", "project", "uuid", "body"]
+        _all_params.extend(
+            [
+                "async_req",
+                "_return_http_data_only",
+                "_preload_content",
+                "_request_timeout",
+                "_request_auth",
+                "_content_type",
+                "_headers",
+            ]
+        )
+
+        # validate the arguments
+        for _key, _val in _params["kwargs"].items():
+            if _key not in _all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method set_run_edges_lineage" % _key
+                )
+            _params[_key] = _val
+        del _params["kwargs"]
+
+        _collection_formats = {}
+
+        # process the path parameters
+        _path_params = {}
+        if _params["owner"]:
+            _path_params["owner"] = _params["owner"]
+        if _params["project"]:
+            _path_params["project"] = _params["project"]
+        if _params["uuid"]:
+            _path_params["uuid"] = _params["uuid"]
+
+        # process the query parameters
+        _query_params = []
+        # process the header parameters
+        _header_params = dict(_params.get("_headers", {}))
+        # process the form parameters
+        _form_params = []
+        _files = {}
+        # process the body parameter
+        _body_params = None
+        if _params["body"]:
+            _body_params = _params["body"]
+
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json"]
+        )  # noqa: E501
+
+        # set the HTTP header `Content-Type`
+        _content_types_list = _params.get(
+            "_content_type",
+            self.api_client.select_header_content_type(["application/json"]),
+        )
+        if _content_types_list:
+            _header_params["Content-Type"] = _content_types_list
+
+        # authentication setting
+        _auth_settings = ["ApiKey"]  # noqa: E501
+
+        _response_types_map = {}
+
+        return self.api_client.call_api(
+            "/api/v1/{owner}/{project}/runs/{uuid}/lineage/edges",
+            "POST",
+            _path_params,
+            _query_params,
+            _header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            response_types_map=_response_types_map,
+            auth_settings=_auth_settings,
+            async_req=_params.get("async_req"),
+            _return_http_data_only=_params.get("_return_http_data_only"),  # noqa: E501
+            _preload_content=_params.get("_preload_content", True),
+            _request_timeout=_params.get("_request_timeout"),
+            collection_formats=_collection_formats,
+            _request_auth=_params.get("_request_auth"),
+        )
+
     @validate_arguments
     def stop_run(
         self,

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_sdk/schemas/__init__.py

@@ -69,6 +69,8 @@ from polyaxon._sdk.schemas.v1_queue import V1Queue
 from polyaxon._sdk.schemas.v1_run import V1Run
 from polyaxon._sdk.schemas.v1_run_connection import V1RunConnection
 from polyaxon._sdk.schemas.v1_run_edge import V1RunEdge
+from polyaxon._sdk.schemas.v1_run_edge_lineage import V1RunEdgeLineage
+from polyaxon._sdk.schemas.v1_run_edges_graph import V1RunEdgesGraph
 from polyaxon._sdk.schemas.v1_run_reference_catalog import V1RunReferenceCatalog
 from polyaxon._sdk.schemas.v1_run_settings import V1RunSettings
 from polyaxon._sdk.schemas.v1_search import V1Search

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/_sdk/schemas/v1_run.py

@@ -32,8 +32,8 @@ class V1Run(BaseAllowSchemaModel):
     updated_at: Optional[datetime.datetime]
     started_at: Optional[datetime.datetime]
     finished_at: Optional[datetime.datetime]
-    wait_time: Optional[int]
-    duration: Optional[int]
+    wait_time: Optional[float]
+    duration: Optional[float]
     managed_by: Optional[ManagedBy]
     is_managed: Optional[bool]
     is_approved: Optional[bool]

polyaxon-2.0.0rc51/polyaxon/_sdk/schemas/v1_run_edge_lineage.py

@@ -0,0 +1,14 @@
+from typing import Any, Dict, Optional
+
+from clipped.config.schema import BaseAllowSchemaModel
+from clipped.types.uuids import UUIDStr
+
+
+class V1RunEdgeLineage(BaseAllowSchemaModel):
+    """
+    V1RunEdgeLineage
+    """
+
+    uuid: Optional[UUIDStr]
+    is_upstream: Optional[bool]
+    values: Optional[Dict[str, Any]]

polyaxon-2.0.0rc51/polyaxon/_sdk/schemas/v1_run_edges_graph.py

@@ -0,0 +1,9 @@
+from typing import List, Optional
+
+from clipped.config.schema import BaseAllowSchemaModel
+
+from polyaxon._sdk.schemas.v1_run_edge_lineage import V1RunEdgeLineage
+
+
+class V1RunEdgesGraph(BaseAllowSchemaModel):
+    edges: Optional[List[V1RunEdgeLineage]]

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon/pkg.py

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

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyaxon
-Version: 2.0.0rc50
+Version: 2.0.0rc51
 Summary: Command Line Interface (CLI) and client to interact with Polyaxon API.
 Home-page: https://github.com/polyaxon/polyaxon
 Author: Polyaxon, Inc.

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon.egg-info/SOURCES.txt

@@ -528,6 +528,8 @@ polyaxon/_sdk/schemas/v1_queue.py
 polyaxon/_sdk/schemas/v1_run.py
 polyaxon/_sdk/schemas/v1_run_connection.py
 polyaxon/_sdk/schemas/v1_run_edge.py
+polyaxon/_sdk/schemas/v1_run_edge_lineage.py
+polyaxon/_sdk/schemas/v1_run_edges_graph.py
 polyaxon/_sdk/schemas/v1_run_reference_catalog.py
 polyaxon/_sdk/schemas/v1_run_settings.py
 polyaxon/_sdk/schemas/v1_search.py

{polyaxon-2.0.0rc50 → polyaxon-2.0.0rc51}/polyaxon.egg-info/requires.txt

@@ -15,8 +15,8 @@ sentry-sdk>=1.2.0
 urllib3>=1.25.6
 certifi>=2022.12.7
 pydantic>=1.10.2
-traceml==1.1.0rc41
-hypertune==1.1.0rc41
+traceml==1.1.0rc42
+hypertune==1.1.0rc42
 
 [azure]
 adlfs