polyaxon 2.1.0rc1__py3-none-any.whl → 2.1.0rc3__py3-none-any.whl

polyaxon/_k8s/manager/async_manager.py

@@ -59,7 +59,9 @@ class AsyncK8sManager(BaseK8sManager):
                 raise PolyaxonK8sError("Connection error: %s" % e) from e
             return None
 
-    async def is_pod_running(self, pod_id: str, container_id: str, namespace: str = None) -> bool:
+    async def is_pod_running(
+        self, pod_id: str, container_id: str, namespace: str = None
+    ) -> bool:
         event = await self.k8s_api.read_namespaced_pod_status(pod_id, namespace=namespace or self.namespace)  # type: ignore[attr-defined]
         return is_pod_running(event, container_id)
 

polyaxon/_runner/agent/async_agent.py

@@ -197,12 +197,13 @@ class BaseAsyncAgent(BaseAgent):
         if not resource:
             return
 
+        namespace = None if len(run_data) < 5 else run_data[4]
         try:
             await self.executor.create(
                 run_uuid=run_uuid,
                 run_kind=run_data[1],
                 resource=resource,
-                namespace=run_data[4],
+                namespace=namespace,
             )
         except ApiException as e:
             if e.status == 409:
@@ -239,12 +240,14 @@ class BaseAsyncAgent(BaseAgent):
         if not resource:
             return
 
+        namepsace = None if len(run_data) < 5 else run_data[4]
+
         try:
             await self.executor.create(
                 run_uuid=run_uuid,
                 run_kind=run_data[1],
                 resource=resource,
-                namespace=run_data[4],
+                namespace=namepsace,
             )
         except ApiException as e:
             if e.status == 409:
@@ -270,12 +273,14 @@ class BaseAsyncAgent(BaseAgent):
         if not resource:
             return
 
+        namespace = None if len(run_data) < 5 else run_data[4]
+
         try:
             await self.executor.apply(
                 run_uuid=run_uuid,
                 run_kind=run_data[1],
                 resource=resource,
-                namespace=run_data[4],
+                namespace=namespace,
             )
             await self.client.log_run_running(
                 run_owner=run_owner, run_project=run_project, run_uuid=run_uuid
@@ -285,14 +290,15 @@ class BaseAsyncAgent(BaseAgent):
                 run_owner=run_owner, run_project=run_project, run_uuid=run_uuid, exc=e
             )
             await self.clean_run(
-                run_uuid=run_uuid, run_kind=run_data[1], namespace=run_data[4]
+                run_uuid=run_uuid, run_kind=run_data[1], namespace=namespace
             )
 
     async def check_run(self, run_data: Tuple[str, str, str]):
         run_owner, run_project, run_uuid = get_run_info(run_instance=run_data[0])
+        namespace = None if len(run_data) < 3 else run_data[2]
         try:
             await self.executor.get(
-                run_uuid=run_uuid, run_kind=run_data[1], namespace=run_data[2]
+                run_uuid=run_uuid, run_kind=run_data[1], namespace=namespace
             )
         except ApiException as e:
             if e.status == 404:
@@ -305,9 +311,10 @@ class BaseAsyncAgent(BaseAgent):
 
     async def stop_run(self, run_data: Tuple[str, str, str]):
         run_owner, run_project, run_uuid = get_run_info(run_instance=run_data[0])
+        namespace = None if len(run_data) < 3 else run_data[2]
         try:
             await self.executor.stop(
-                run_uuid=run_uuid, run_kind=run_data[1], namespace=run_data[2]
+                run_uuid=run_uuid, run_kind=run_data[1], namespace=namespace
             )
         except ApiException as e:
             if e.status == 404:
@@ -326,11 +333,12 @@ class BaseAsyncAgent(BaseAgent):
 
     async def delete_run(self, run_data: Tuple[str, str, str, str, str]):
         run_owner, run_project, run_uuid = get_run_info(run_instance=run_data[0])
+        namespace = None if len(run_data) < 5 else run_data[4]
         if run_data[3]:
             await self.make_and_create_run(run_data)
         else:
             await self.clean_run(
-                run_uuid=run_uuid, run_kind=run_data[1], namespace=run_data[4]
+                run_uuid=run_uuid, run_kind=run_data[1], namespace=namespace
             )
 
     async def clean_run(self, run_uuid: str, run_kind: str, namespace: str = None):

polyaxon/_sdk/api/agents_v1_api.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Dict, Optional
 from typing_extensions import Annotated
 
 from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_arguments
@@ -2399,6 +2399,7 @@ class AgentsV1Api(BaseApi):
     def cron_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
+        body: Annotated[Optional[Dict], Field(..., description="Cron body")] = None,
         **kwargs
     ) -> V1AgentStateResponse:  # noqa: E501
         """Global Cron agent  # noqa: E501
@@ -2406,11 +2407,13 @@ class AgentsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.cron_agent(owner, async_req=True)
+        >>> thread = api.cron_agent(owner, body, async_req=True)
         >>> result = thread.get()
 
         :param owner: Owner of the namespace (required)
         :type owner: str
+        :param body: Cron body (required)
+        :type body: object
         :param async_req: Whether to execute the request asynchronously.
         :type async_req: bool, optional
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -2427,12 +2430,13 @@ class AgentsV1Api(BaseApi):
         :rtype: V1AgentStateResponse
         """
         kwargs["_return_http_data_only"] = True
-        return self.cron_agent_with_http_info(owner, **kwargs)  # noqa: E501
+        return self.cron_agent_with_http_info(owner, body, **kwargs)  # noqa: E501
 
     @validate_arguments
     def cron_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
+        body: Annotated[Optional[Dict], Field(..., description="Cron body")] = None,
         **kwargs
     ):  # noqa: E501
         """Cron agent  # noqa: E501
@@ -2440,11 +2444,13 @@ class AgentsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.cron_agent_with_http_info(owner, async_req=True)
+        >>> thread = api.cron_agent_with_http_info(owner, body, async_req=True)
         >>> result = thread.get()
 
         :param owner: Owner of the namespace (required)
         :type owner: str
+        :param body: Cron body (required)
+        :type body: object
         :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
@@ -2471,7 +2477,7 @@ class AgentsV1Api(BaseApi):
 
         _params = locals()
 
-        _all_params = ["owner"]
+        _all_params = ["owner", "body"]
         _all_params.extend(
             [
                 "async_req",
@@ -2510,11 +2516,21 @@ class AgentsV1Api(BaseApi):
         _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
 

polyaxon/pkg.py

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

{polyaxon-2.1.0rc1.dist-info → polyaxon-2.1.0rc3.dist-info}/METADATA

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

{polyaxon-2.1.0rc1.dist-info → polyaxon-2.1.0rc3.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=_AjVHLBB1kJdzQbcjijfbbYvdHWpCCAX40TA44wZtWw,266
+polyaxon/pkg.py,sha256=pp7j7yEBC1bpoS3D0J4SHsSAGUZ9E6GGygvSHN_FPvk,266
 polyaxon/polyaxonfile.py,sha256=xHmHT_cHomfuAQm82Jhnp71YNN5mQ-Lod7EbonjY4b4,429
 polyaxon/schemas.py,sha256=l5Lg1IGP6FM_pO8eToY0RUCzVD1kn3Dk5CNM_nsAf60,5877
 polyaxon/settings.py,sha256=Pxx1-T2oeJ5XmvGFN0YgnVzum_9FyTPaQtl68aQvYc4,4116
@@ -345,7 +345,7 @@ polyaxon/_k8s/logging/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdr
 polyaxon/_k8s/logging/async_monitor.py,sha256=fXOjl7mkRpGw4MsMJgICYBGV_27DjJf2fjlEY-AOQxw,3886
 polyaxon/_k8s/logging/monitor.py,sha256=4-NLN4_y0QVsReNJr8Ekv8kh0XNRRTKiiHopWs4wtMc,3426
 polyaxon/_k8s/manager/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-polyaxon/_k8s/manager/async_manager.py,sha256=MeYa019OkOl9L4ykdHUu4223t9W291nYJ3AI1fhoO30,8850
+polyaxon/_k8s/manager/async_manager.py,sha256=3mnc5iVmoD4xmdw3CgJkVS3wdnN6TTvMbWz3Wu3BOAg,8864
 polyaxon/_k8s/manager/base.py,sha256=hCe6RwRlO5ZYjz7P5fEoIHrrhf1yIof7DCzaEqM4kxw,2727
 polyaxon/_k8s/manager/manager.py,sha256=eJESZ6upQjeUrAYeU13cwXb_jmDYp56_SoysCZbY7VU,30143
 polyaxon/_managers/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
@@ -394,7 +394,7 @@ polyaxon/_runner/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,
 polyaxon/_runner/executor.py,sha256=ZxH_NxSrzDj_tHTCVmZ3MiXXFzKQdL_mke2GVp6229I,8175
 polyaxon/_runner/kinds.py,sha256=IhEL-HfztpDU4wUA7z8koQwN1Z32g2q57iDiiT9YZ40,132
 polyaxon/_runner/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-polyaxon/_runner/agent/async_agent.py,sha256=W_UCpX9N_s4wSXXk8m39EybrM5IYQNQrOG3ihBaS_rY,13346
+polyaxon/_runner/agent/async_agent.py,sha256=8XQ-J89vsrirTmVHf2Q7wjcvPfaAPuF3a4xoI-bE_KY,13712
 polyaxon/_runner/agent/base_agent.py,sha256=mBMTEqjbbqxV2O23vwUSRNVKyRJN-dChqtMNNSXy_Ac,6709
 polyaxon/_runner/agent/client.py,sha256=iZDPH97qKGIcEZwHwqYPeWrcWp8Tpo617MwK9_tSnkM,5035
 polyaxon/_runner/agent/sync_agent.py,sha256=9dI0S8KI85mubAD9JRLHtw9QPecnGZouMcX95N2GHgA,12891
@@ -441,7 +441,7 @@ polyaxon/_sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 polyaxon/_sdk/base_api.py,sha256=7LXnjh3jGHWqIMFj81Df21Z0nAhDIQqm5TALvMj15eU,393
 polyaxon/_sdk/configuration.py,sha256=QDZ9grJjtHuL7TQhWZqbAcBFlh9674go6kxaQ10XCWo,15103
 polyaxon/_sdk/api/__init__.py,sha256=ABYkNzoTkFk83AGh5eHBuNtwUdbJt9kNURKr9wzYSzk,1118
-polyaxon/_sdk/api/agents_v1_api.py,sha256=5uFK2FXFm3L9vpFx4IcR_IRWG-bmQheG8OS_xSGn9vQ,122436
+polyaxon/_sdk/api/agents_v1_api.py,sha256=CeDSPKK0T8VytyElAQUny39HXkR0F7PrLmuBg9xFzQ4,123150
 polyaxon/_sdk/api/artifacts_stores_v1_api.py,sha256=VuBdfwKi0URQQCuh9GeHceRPMc7fOlJYl7-K-okJtH4,8403
 polyaxon/_sdk/api/auth_v1_api.py,sha256=Bov8wCCYmv8r1cZyodmIyDC_SahUpMsTKQsOa-vbuyg,35802
 polyaxon/_sdk/api/connections_v1_api.py,sha256=PPTGX7FrmD1zls1lWBukCWgNuRn7Eui365Ji5t_Sz0A,54365
@@ -593,9 +593,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.1.0rc1.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
-polyaxon-2.1.0rc1.dist-info/METADATA,sha256=7ICMEzbhernXUGamPBl9KINrFO9Bo5YRDs-f5pWztu8,11716
-polyaxon-2.1.0rc1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-polyaxon-2.1.0rc1.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
-polyaxon-2.1.0rc1.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
-polyaxon-2.1.0rc1.dist-info/RECORD,,
+polyaxon-2.1.0rc3.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
+polyaxon-2.1.0rc3.dist-info/METADATA,sha256=jwaUieEKgXxyP1je6ESxeF-xqdTLq9LHgW_Xw1ghnqw,11716
+polyaxon-2.1.0rc3.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+polyaxon-2.1.0rc3.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
+polyaxon-2.1.0rc3.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
+polyaxon-2.1.0rc3.dist-info/RECORD,,