polyaxon 2.4.0rc1__py3-none-any.whl → 2.6.0__py3-none-any.whl

polyaxon/_schemas/services.py

@@ -1,6 +1,12 @@
 from typing import Dict, List, Optional, Union
 
-from clipped.compact.pydantic import Field, StrictStr, validator
+from clipped.compact.pydantic import (
+    Field,
+    StrictStr,
+    field_validator,
+    validation_always,
+    validation_before,
+)
 
 from polyaxon._containers.pull_policy import PullPolicy
 from polyaxon._k8s import k8s_schemas, k8s_validation
@@ -14,25 +20,31 @@ class BaseServiceConfig(BaseSchemaModel):
         "tolerations",
     }
 
-    image: Optional[StrictStr]
-    image_tag: Optional[StrictStr] = Field(alias="imageTag")
-    image_pull_policy: Optional[PullPolicy] = Field(alias="imagePullPolicy")
-    resources: Optional[Union[k8s_schemas.V1ResourceRequirements, Dict]]
-    node_selector: Optional[Dict[StrictStr, StrictStr]] = Field(alias="nodeSelector")
-    affinity: Optional[Union[k8s_schemas.V1Affinity, Dict]]
-    tolerations: Optional[List[Union[k8s_schemas.V1Toleration, Dict]]]
-    image_pull_secrets: Optional[List[StrictStr]] = Field(alias="imagePullSecrets")
-    hpa: Optional[Dict]
-
-    @validator("resources", always=True, pre=True)
+    image: Optional[StrictStr] = None
+    image_tag: Optional[StrictStr] = Field(default=None, alias="imageTag")
+    image_pull_policy: Optional[PullPolicy] = Field(
+        default=None, alias="imagePullPolicy"
+    )
+    resources: Optional[Union[k8s_schemas.V1ResourceRequirements, Dict]] = None
+    node_selector: Optional[Dict[StrictStr, StrictStr]] = Field(
+        default=None, alias="nodeSelector"
+    )
+    affinity: Optional[Union[k8s_schemas.V1Affinity, Dict]] = None
+    tolerations: Optional[List[Union[k8s_schemas.V1Toleration, Dict]]] = None
+    image_pull_secrets: Optional[List[StrictStr]] = Field(
+        default=None, alias="imagePullSecrets"
+    )
+    hpa: Optional[Dict] = None
+
+    @field_validator("resources", **validation_always, **validation_before)
     def validate_resources(cls, v):
         return k8s_validation.validate_k8s_resource_requirements(v)
 
-    @validator("affinity", always=True, pre=True)
+    @field_validator("affinity", **validation_always, **validation_before)
     def validate_affinity(cls, v):
         return k8s_validation.validate_k8s_affinity(v)
 
-    @validator("tolerations", always=True, pre=True)
+    @field_validator("tolerations", **validation_always, **validation_before)
     def validate_tolerations(cls, v):
         if not v:
             return v

polyaxon/_schemas/types/artifacts.py

@@ -154,6 +154,6 @@ class V1ArtifactsType(BaseTypeConfig):
 
     _IDENTIFIER = "artifacts"
 
-    files: Optional[Union[List[Union[str, List[str]]], RefField]]
-    dirs: Optional[Union[List[Union[str, List[str]]], RefField]]
-    workers: Optional[Union[StrictInt, RefField]]
+    files: Optional[Union[List[Union[str, List[str]]], RefField]] = None
+    dirs: Optional[Union[List[Union[str, List[str]]], RefField]] = None
+    workers: Optional[Union[StrictInt, RefField]] = None

polyaxon/_schemas/types/dockerfile.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-from clipped.compact.pydantic import Field, StrictStr, validator
+from clipped.compact.pydantic import Field, StrictStr, field_validator
 from clipped.config.schema import skip_partial
 from clipped.types.docker_image import validate_image
 from clipped.types.ref_or_obj import RefField
@@ -179,31 +179,33 @@ class V1DockerfileType(BaseTypeConfig):
     _IDENTIFIER = "dockerfile"
 
     image: StrictStr
-    env: Optional[Union[Dict[StrictStr, Any], RefField]]
-    path: Optional[Union[List[StrictStr], RefField]]
+    env: Optional[
+        Union[Dict[str, Any], List[Union[Tuple[str, str], List[str]]], RefField]
+    ] = None
+    path: Optional[Union[List[StrictStr], RefField]] = None
     copy_: Optional[
         Union[
             List[Union[StrictStr, List[StrictStr], Tuple[StrictStr, StrictStr]]],
             RefField,
         ]
-    ] = Field(alias="copy")
+    ] = Field(alias="copy", default=None)
     post_run_copy: Optional[
         Union[
             List[Union[StrictStr, List[StrictStr], Tuple[StrictStr, StrictStr]]],
             RefField,
         ]
-    ] = Field(alias="postRunCopy")
-    run: Optional[Union[List[StrictStr], RefField]]
-    lang_env: Optional[StrictStr] = Field(alias="langEnv")
-    uid: Optional[Union[int, RefField]]
-    gid: Optional[Union[int, RefField]]
-    username: Optional[StrictStr]
+    ] = Field(alias="postRunCopy", default=None)
+    run: Optional[Union[List[StrictStr], RefField]] = None
+    lang_env: Optional[StrictStr] = Field(alias="langEnv", default=None)
+    uid: Optional[Union[int, RefField]] = None
+    gid: Optional[Union[int, RefField]] = None
+    username: Optional[StrictStr] = None
     filename: Optional[StrictStr] = Field(default=POLYAXON_DOCKERFILE_NAME)
     workdir: Optional[StrictStr] = Field(default=POLYAXON_DOCKER_WORKDIR)
-    workdir_path: Optional[StrictStr] = Field(alias="workdirPath")
+    workdir_path: Optional[StrictStr] = Field(alias="workdirPath", default=None)
     shell: Optional[StrictStr] = Field(default=POLYAXON_DOCKER_SHELL)
 
-    @validator("image")
+    @field_validator("image")
     @skip_partial
     def check_image(cls, image):
         validate_image(image)

polyaxon/_schemas/types/event.py

@@ -8,5 +8,5 @@ from polyaxon._schemas.types.base import BaseTypeConfig
 class V1EventType(BaseTypeConfig):
     _IDENTIFIER = "event"
 
-    name: Optional[StrictStr]
-    kind: Optional[StrictStr]
+    name: Optional[StrictStr] = None
+    kind: Optional[StrictStr] = None

polyaxon/_schemas/types/file.py

@@ -143,7 +143,7 @@ class V1FileType(BaseTypeConfig):
 
     _IDENTIFIER = "file"
 
-    kind: Optional[Union[V1ArtifactKind, RefField]]
+    kind: Optional[Union[V1ArtifactKind, RefField]] = None
     content: StrictStr
-    filename: Optional[StrictStr]
-    chmod: Optional[StrictStr]
+    filename: Optional[StrictStr] = None
+    chmod: Optional[StrictStr] = None

polyaxon/_schemas/types/git.py

@@ -1,10 +1,13 @@
-from typing import List, Optional, Union
+from typing import TYPE_CHECKING, List, Optional, Union
 
 from clipped.compact.pydantic import StrictStr
 from clipped.types.ref_or_obj import RefField
 
 from polyaxon._schemas.types.base import BaseTypeConfig
 
+if TYPE_CHECKING:
+    from polyaxon._connections import V1GitConnection
+
 
 class V1GitType(BaseTypeConfig):
     """Git type allows you to pass a git repo as a parameter.
@@ -124,11 +127,16 @@ class V1GitType(BaseTypeConfig):
 
     _IDENTIFIER = "git"
 
-    url: Optional[StrictStr]
-    revision: Optional[StrictStr]
-    flags: Optional[Union[List[StrictStr], RefField]]
+    url: Optional[StrictStr] = None
+    revision: Optional[StrictStr] = None
+    flags: Optional[Union[List[StrictStr], RefField]] = None
 
     def get_name(self):
         if self.url:
             return self.url.split("/")[-1].split(".")[0]
         return None
+
+    def to_connection(self) -> "V1GitConnection":
+        from polyaxon._connections import V1GitConnection
+
+        return V1GitConnection(url=self.url, revision=self.revision, flags=self.flags)

polyaxon/_schemas/types/tensorboard.py

@@ -1,6 +1,12 @@
 from typing import List, Optional, Union
 
-from clipped.compact.pydantic import Field, StrictInt, StrictStr, validator
+from clipped.compact.pydantic import (
+    Field,
+    StrictInt,
+    StrictStr,
+    field_validator,
+    validation_before,
+)
 from clipped.config.constants import PARAM_REGEX
 from clipped.types.ref_or_obj import RefField
 from clipped.types.uuids import UUIDStr
@@ -129,14 +135,14 @@ class V1TensorboardType(BaseTypeConfig):
 
     _IDENTIFIER = "tensorboard"
 
-    port: Optional[Union[StrictInt, RefField]]
-    uuids: Optional[Union[List[UUIDStr], RefField]]
-    use_names: Optional[Union[bool, RefField]] = Field(alias="useNames")
-    path_prefix: Optional[StrictStr] = Field(alias="pathPrefix")
-    plugins: Optional[Union[List[StrictStr], RefField]]
+    port: Optional[Union[StrictInt, RefField]] = None
+    uuids: Optional[Union[List[UUIDStr], RefField]] = None
+    use_names: Optional[Union[bool, RefField]] = Field(alias="useNames", default=None)
+    path_prefix: Optional[StrictStr] = Field(alias="pathPrefix", default=None)
+    plugins: Optional[Union[List[StrictStr], RefField]] = None
 
-    @validator("uuids", "plugins", pre=True)
-    def validate_str_list(cls, v, field):
+    @field_validator("uuids", "plugins", **validation_before)
+    def validate_str_list(cls, v):
         if isinstance(v, str) and v is not None and not PARAM_REGEX.search(v):
             return to_list(v, check_str=True)
         return v

polyaxon/_schemas/user.py

@@ -10,6 +10,6 @@ class UserConfig(BaseSchemaModel):
     _IDENTIFIER = "user"
 
     username: StrictStr
-    email: Optional[EmailStr]
-    name: Optional[StrictStr]
-    theme: Optional[int]
+    email: Optional[EmailStr] = None
+    name: Optional[StrictStr] = None
+    theme: Optional[int] = None

polyaxon/_schemas/version.py

@@ -6,5 +6,5 @@ from polyaxon._schemas.base import BaseSchemaModel
 
 
 class V1Version(BaseSchemaModel):
-    min: Optional[StrictStr]
-    latest: Optional[StrictStr]
+    min: Optional[StrictStr] = None
+    latest: Optional[StrictStr] = None

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,7 @@ 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")],
@@ -1515,7 +1515,7 @@ class AgentsV1Api(BaseApi):
             owner, uuid, entity, **kwargs
         )  # noqa: E501
 
-    @validate_arguments
+    @validate_call
     def get_agent_statuses_with_http_info(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1647,7 +1647,7 @@ class AgentsV1Api(BaseApi):
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
+    @validate_call
     def get_agent_token(
         self,
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1686,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")],
@@ -1812,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")],
@@ -1846,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")],
@@ -1961,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")],
@@ -2023,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")],
@@ -2185,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")],
@@ -2252,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")],
@@ -2425,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")],
@@ -2492,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")],
@@ -2665,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")],
@@ -2707,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")],
@@ -2844,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")],
@@ -2886,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")],
@@ -3023,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")],
@@ -3067,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")],
@@ -3203,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")],
@@ -3245,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")],
@@ -3377,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")],
@@ -3414,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")],
@@ -3542,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")],
@@ -3584,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")],
@@ -3721,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")],
@@ -3763,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")],
@@ -3900,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")],
@@ -3942,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")],