polyaxon 2.1.0rc9__py3-none-any.whl → 2.6.0__py3-none-any.whl

polyaxon/_sdk/api/agents_v1_api.py

@@ -1,7 +1,7 @@
 from typing import Dict, Optional
 from typing_extensions import Annotated
 
-from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_arguments
+from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_call
 
 from polyaxon._schemas.lifecycle import V1Status
 from polyaxon._sdk.base_api import BaseApi
@@ -18,7 +18,7 @@ from traceml.logging import V1Logs
 
 
 class AgentsV1Api(BaseApi):
-    @validate_arguments
+    @validate_call
     def collect_agent_data(
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
@@ -62,7 +62,7 @@ class AgentsV1Api(BaseApi):
             namespace, owner, uuid, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def collect_agent_data_with_http_info(
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
@@ -189,7 +189,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def create_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -226,7 +226,7 @@ class AgentsV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.create_agent_with_http_info(owner, body, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def create_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -358,7 +358,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def create_agent_status(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -402,7 +402,7 @@ class AgentsV1Api(BaseApi):
             owner, uuid, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def create_agent_status_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -541,7 +541,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def delete_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -588,7 +588,7 @@ class AgentsV1Api(BaseApi):
             owner, uuid, entity, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def delete_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -717,7 +717,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def get_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -764,7 +764,7 @@ class AgentsV1Api(BaseApi):
             owner, uuid, entity, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def get_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -898,7 +898,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def get_agent_config(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -945,7 +945,7 @@ class AgentsV1Api(BaseApi):
             owner, uuid, entity, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def get_agent_config_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1079,7 +1079,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def get_agent_logs(
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
@@ -1141,7 +1141,7 @@ class AgentsV1Api(BaseApi):
             namespace, owner, uuid, service, last_file, force, connection, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def get_agent_logs_with_http_info(
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
@@ -1303,7 +1303,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def get_agent_state(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1342,7 +1342,7 @@ class AgentsV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.get_agent_state_with_http_info(owner, uuid, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def get_agent_state_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1468,7 +1468,186 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
+    def get_agent_statuses(
+        self,
+        owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
+        uuid: Annotated[
+            StrictStr, Field(..., description="Uuid identifier of the sub-entity")
+        ],
+        entity: Annotated[
+            Optional[StrictStr],
+            Field(description="Entity: project name, hub name, registry name, ..."),
+        ] = None,
+        **kwargs
+    ) -> V1Status:  # noqa: E501
+        """Get Agent statuses  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.get_agent_statuses(owner, uuid, entity, async_req=True)
+        >>> result = thread.get()
+
+        :param owner: Owner of the namespace (required)
+        :type owner: str
+        :param uuid: Uuid identifier of the sub-entity (required)
+        :type uuid: str
+        :param entity: Entity: project name, hub name, registry name, ...
+        :type entity: str
+        :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: V1Status
+        """
+        kwargs["_return_http_data_only"] = True
+        return self.get_agent_statuses_with_http_info(
+            owner, uuid, entity, **kwargs
+        )  # noqa: E501
+
+    @validate_call
+    def get_agent_statuses_with_http_info(
+        self,
+        owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
+        uuid: Annotated[
+            StrictStr, Field(..., description="Uuid identifier of the sub-entity")
+        ],
+        entity: Annotated[
+            Optional[StrictStr],
+            Field(description="Entity: project name, hub name, registry name, ..."),
+        ] = None,
+        **kwargs
+    ):  # noqa: E501
+        """Get Agent statuses  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.get_agent_statuses_with_http_info(owner, uuid, entity, async_req=True)
+        >>> result = thread.get()
+
+        :param owner: Owner of the namespace (required)
+        :type owner: str
+        :param uuid: Uuid identifier of the sub-entity (required)
+        :type uuid: str
+        :param entity: Entity: project name, hub name, registry name, ...
+        :type entity: str
+        :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: tuple(V1Status, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["owner", "uuid", "entity"]
+        _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 get_agent_statuses" % _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["uuid"]:
+            _path_params["uuid"] = _params["uuid"]
+
+        # process the query parameters
+        _query_params = []
+        if _params.get("entity") is not None:  # noqa: E501
+            _query_params.append(("entity", _params["entity"]))
+
+        # process the header parameters
+        _header_params = dict(_params.get("_headers", {}))
+        # process the form parameters
+        _form_params = []
+        _files = {}
+        # process the body parameter
+        _body_params = None
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json"]
+        )  # noqa: E501
+
+        # authentication setting
+        _auth_settings = ["ApiKey"]  # noqa: E501
+
+        _response_types_map = {
+            "200": "V1Status",
+            "204": "object",
+            "403": "object",
+            "404": "object",
+        }
+
+        return self.api_client.call_api(
+            "/api/v1/orgs/{owner}/agents/{uuid}/statuses",
+            "GET",
+            _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_call
     def get_agent_token(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1507,7 +1686,7 @@ class AgentsV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.get_agent_token_with_http_info(owner, uuid, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def get_agent_token_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1633,7 +1812,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def get_global_state(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1667,7 +1846,7 @@ class AgentsV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.get_global_state_with_http_info(owner, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def get_global_state_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1782,7 +1961,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def inspect_agent(
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
@@ -1844,7 +2023,7 @@ class AgentsV1Api(BaseApi):
             namespace, owner, uuid, service, last_file, force, connection, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def inspect_agent_with_http_info(
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
@@ -2006,7 +2185,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def list_agent_names(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2073,7 +2252,7 @@ class AgentsV1Api(BaseApi):
             owner, offset, limit, sort, query, bookmarks, mode, no_page, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def list_agent_names_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2246,7 +2425,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def list_agents(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2313,7 +2492,7 @@ class AgentsV1Api(BaseApi):
             owner, offset, limit, sort, query, bookmarks, mode, no_page, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def list_agents_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2486,7 +2665,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def patch_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2528,7 +2707,7 @@ class AgentsV1Api(BaseApi):
             owner, agent_uuid, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def patch_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2665,7 +2844,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def patch_agent_token(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2707,7 +2886,7 @@ class AgentsV1Api(BaseApi):
             owner, entity, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def patch_agent_token_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2844,7 +3023,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def reconcile_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -2888,7 +3067,7 @@ class AgentsV1Api(BaseApi):
             owner, uuid, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def reconcile_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3024,7 +3203,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def sync_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3066,7 +3245,7 @@ class AgentsV1Api(BaseApi):
             owner, agent_uuid, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def sync_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3198,7 +3377,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def cron_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3235,7 +3414,7 @@ class AgentsV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.cron_agent_with_http_info(owner, body, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def cron_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3363,7 +3542,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def update_agent(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3405,7 +3584,7 @@ class AgentsV1Api(BaseApi):
             owner, agent_uuid, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def update_agent_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3542,7 +3721,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def update_agent_config(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3584,7 +3763,7 @@ class AgentsV1Api(BaseApi):
             owner, agent_uuid, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def update_agent_config_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3721,7 +3900,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def update_agent_token(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -3763,7 +3942,7 @@ class AgentsV1Api(BaseApi):
             owner, entity, body, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def update_agent_token_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],

polyaxon/_sdk/api/artifacts_stores_v1_api.py

@@ -1,14 +1,14 @@
 from typing import Optional
 from typing_extensions import Annotated
 
-from clipped.compact.pydantic import Field, StrictStr, validate_arguments
+from clipped.compact.pydantic import Field, StrictStr, validate_call
 
 from polyaxon._sdk.base_api import BaseApi
 from polyaxon.exceptions import ApiTypeError
 
 
 class ArtifactsStoresV1Api(BaseApi):
-    @validate_arguments
+    @validate_call
     def upload_artifact(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -62,7 +62,7 @@ class ArtifactsStoresV1Api(BaseApi):
             owner, uuid, uploadfile, path, overwrite, **kwargs
         )
 
-    @validate_arguments
+    @validate_call
     def upload_artifact_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],

polyaxon/_sdk/api/auth_v1_api.py

@@ -1,4 +1,4 @@
-from clipped.compact.pydantic import validate_arguments
+from clipped.compact.pydantic import validate_call
 
 from polyaxon._schemas.authentication import V1Credentials
 from polyaxon._sdk.base_api import BaseApi
@@ -11,7 +11,7 @@ from polyaxon.exceptions import ApiTypeError
 
 
 class AuthV1Api(BaseApi):
-    @validate_arguments
+    @validate_call
     def change_password(self, body: V1PasswordChange, **kwargs) -> None:
         """Change password
 
@@ -41,7 +41,7 @@ class AuthV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.change_password_with_http_info(body, **kwargs)
 
-    @validate_arguments
+    @validate_call
     def change_password_with_http_info(self, body: V1PasswordChange, **kwargs):
         """Change password
 
@@ -159,7 +159,7 @@ class AuthV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def login(self, body: V1Credentials, **kwargs) -> V1Auth:  # noqa: E501
         """Login  # noqa: E501
 
@@ -189,7 +189,7 @@ class AuthV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.login_with_http_info(body, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def login_with_http_info(self, body: V1Credentials, **kwargs):  # noqa: E501
         """Login  # noqa: E501
 
@@ -311,7 +311,7 @@ class AuthV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def reset_password(self, body: V1UserEmail, **kwargs) -> None:  # noqa: E501
         """Reset password  # noqa: E501
 
@@ -341,7 +341,7 @@ class AuthV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.reset_password_with_http_info(body, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def reset_password_with_http_info(self, body: V1UserEmail, **kwargs):  # noqa: E501
         """Reset password  # noqa: E501
 
@@ -459,7 +459,7 @@ class AuthV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def reset_password_confirm(
         self, body: V1PasswordChange, **kwargs
     ) -> V1Auth:  # noqa: E501
@@ -491,7 +491,7 @@ class AuthV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.reset_password_confirm_with_http_info(body, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def reset_password_confirm_with_http_info(
         self, body: V1PasswordChange, **kwargs
     ):  # noqa: E501
@@ -616,7 +616,7 @@ class AuthV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def signup(self, body: V1UserSingup, **kwargs) -> V1Auth:  # noqa: E501
         """Signup  # noqa: E501
 
@@ -646,7 +646,7 @@ class AuthV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.signup_with_http_info(body, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def signup_with_http_info(self, body: V1UserSingup, **kwargs):  # noqa: E501
         """Signup  # noqa: E501
 
@@ -768,7 +768,7 @@ class AuthV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def trial(self, body: V1TrialStart, **kwargs) -> None:  # noqa: E501
         """Trial Start  # noqa: E501
 
@@ -798,7 +798,7 @@ class AuthV1Api(BaseApi):
         kwargs["_return_http_data_only"] = True
         return self.trial_with_http_info(body, **kwargs)  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def trial_with_http_info(self, body: V1TrialStart, **kwargs):  # noqa: E501
         """Trial Start  # noqa: E501