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

polyaxon/_schemas/log_handler.py

@@ -9,8 +9,8 @@ from clipped.config.schema import BaseAllowSchemaModel
 class V1LogHandler(BaseAllowSchemaModel):
     _IDENTIFIER = "log_handler"
 
-    dsn: Optional[StrictStr]
-    environment: Optional[StrictStr]
+    dsn: Optional[StrictStr] = None
+    environment: Optional[StrictStr] = None
 
     @property
     def decoded_dsn(self):

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