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

polyaxon/_cli/operations.py

@@ -1466,7 +1466,7 @@ def logs(ctx, project, uid, follow, hide_time, all_containers, all_info, offline
             )
 
         try:
-            from traceml.logging.streamer import get_logs_streamer, load_logs_from_path
+            from traceml.logging.streamer import load_logs_from_path
 
             load_logs_from_path(
                 logs_path=logs_path,

polyaxon/_cli/projects.py

@@ -46,7 +46,7 @@ def project(ctx, _project):  # pylint:disable=redefined-outer-name
 @click.option(
     *OPTIONS_NAME["args"],
     type=str,
-    help="The project name, e.g. 'mnist' or 'acme/mnist'."
+    help="The project name, e.g. 'mnist' or 'acme/mnist'.",
 )
 @click.option("--description", type=str, help="Description of the project.")
 @click.option("--tags", type=str, help="Tags of the project (comma separated values).")
@@ -287,7 +287,7 @@ def delete(ctx, _project, yes):
 @click.option(
     *OPTIONS_NAME["args"],
     type=str,
-    help="Name of the project, must be unique for the same user."
+    help="Name of the project, must be unique for the same user.",
 )
 @click.option("--description", type=str, help="Description of the project.")
 @click.option("--tags", type=str, help="Tags of the project (comma separated values).")

polyaxon/_compiler/contexts/contexts.py

@@ -98,24 +98,24 @@ def resolve_globals_contexts(
         run_outputs_path = ctx_paths.CONTEXT_MOUNT_RUN_OUTPUTS_FORMAT.format(run_path)
         resolved_contexts[ctx_sections.GLOBALS][
             ctx_keys.RUN_ARTIFACTS_PATH
-        ] = run_artifacts_path
+        ] = run_artifacts_path  # fmt: skip
         resolved_contexts[ctx_sections.GLOBALS][
             ctx_keys.RUN_OUTPUTS_PATH
-        ] = run_outputs_path
+        ] = run_outputs_path  # fmt: skip
     elif artifacts_store:
         run_artifacts_path = os.path.join(artifacts_store.store_path, run_path)
         run_outputs_path = os.path.join(run_artifacts_path, "outputs")
         resolved_contexts[ctx_sections.GLOBALS][
             ctx_keys.RUN_ARTIFACTS_PATH
-        ] = run_artifacts_path
+        ] = run_artifacts_path  # fmt: skip
         resolved_contexts[ctx_sections.GLOBALS][
             ctx_keys.RUN_OUTPUTS_PATH
-        ] = run_outputs_path
+        ] = run_outputs_path  # fmt: skip
 
     if plugins.mount_artifacts_store and artifacts_store:
         resolved_contexts[ctx_sections.GLOBALS][
             ctx_keys.STORE_PATH
-        ] = artifacts_store.store_path
+        ] = artifacts_store.store_path  # fmt: skip
     return resolved_contexts
 
 

polyaxon/_compiler/contexts/dask_job.py

@@ -21,7 +21,7 @@ class DaskJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1DaskJob
+        job: V1DaskJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1DaskReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/contexts/kubeflow/mpi_job.py

@@ -21,7 +21,7 @@ class MPIJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1MPIJob
+        job: V1MPIJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/contexts/kubeflow/mx_job.py

@@ -21,7 +21,7 @@ class MXJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1MXJob
+        job: V1MXJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/contexts/kubeflow/paddle_job.py

@@ -21,7 +21,7 @@ class PaddleJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1PaddleJob
+        job: V1PaddleJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/contexts/kubeflow/pytroch_job.py

@@ -21,7 +21,7 @@ class PytorchJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1PytorchJob
+        job: V1PytorchJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/contexts/kubeflow/tf_job.py

@@ -21,7 +21,7 @@ class TfJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1TFJob
+        job: V1TFJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/contexts/kubeflow/xgb_job.py

@@ -21,7 +21,7 @@ class XGBoostJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1XGBoostJob
+        job: V1XGBoostJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/contexts/ray_job.py

@@ -21,7 +21,7 @@ class RayJobContextsManager(BaseContextsManager):
     ) -> Dict:
         contexts["init"] = {}
         contexts["connections"] = {}
-        job = compiled_operation.run  # type: V1RayJob
+        job: V1RayJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1RayReplica]) -> Dict:
             if not replica:

polyaxon/_compiler/resolver/agent.py

@@ -71,7 +71,7 @@ class AgentResolver(BaseSchemaModel):
         if agent_config.artifacts_store:  # Resolve default artifacts store
             self.connection_by_names[
                 agent_config.artifacts_store.name
-            ] = agent_config.artifacts_store
+            ] = agent_config.artifacts_store  # fmt: skip
 
         if (
             compiled_operation.is_job_run

polyaxon/_compiler/resolver/runtime.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any, Dict, List, Optional
+from typing import Dict, List, Optional
 
 from polyaxon import settings
 from polyaxon._compiler.contexts import resolve_contexts, resolve_globals_contexts

polyaxon/_deploy/schemas/ui.py

@@ -13,4 +13,6 @@ class UIConfig(BaseSchemaModel):
     assets_version: Optional[StrictStr] = Field(alias="assetsVersion", default=None)
     admin_enabled: Optional[bool] = Field(alias="adminEnabled", default=None)
     single_url: Optional[bool] = Field(alias="singleUrl", default=None)
-    default_streams_url: Optional[StrictStr] = Field(alias="defaultStreamsUrl", default=None)
+    default_streams_url: Optional[StrictStr] = Field(
+        alias="defaultStreamsUrl", default=None
+    )

polyaxon/_docker/converter/base/base.py

@@ -162,7 +162,7 @@ class BaseConverter(
         if memory:
             docker_resources[
                 "memory"
-            ] = docker_types.V1ResourceRequirements.from_k8s_memory(memory)
+            ] = docker_types.V1ResourceRequirements.from_k8s_memory(memory)  # fmt: skip
         if gpus:
             docker_resources["gpus"] = gpus
         return docker_types.V1ResourceRequirements.from_dict(docker_resources)

polyaxon/_docker/converter/base/env_vars.py

@@ -98,7 +98,7 @@ class EnvMixin(BaseConverter):
 
         try:
             secret_value = orjson_loads(resource.name)
-        except orjson.JSONDecodeError as e:
+        except orjson.JSONDecodeError:
             return items_from
 
         for item in resource.items:

polyaxon/_docker/converter/converters/job.py

@@ -18,7 +18,7 @@ class JobConverter(JobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> List[docker_types.V1Container]:
-        job = compiled_operation.run  # type: V1Job
+        job: V1Job = compiled_operation.run
         plugins = V1Plugins.get_or_create(
             config=compiled_operation.plugins, auth=default_auth
         )

polyaxon/_docker/converter/converters/service.py

@@ -4,7 +4,7 @@ from polyaxon._connections import V1Connection, V1ConnectionResource
 from polyaxon._docker import docker_types
 from polyaxon._docker.converter.base import BaseConverter
 from polyaxon._docker.converter.mixins import ServiceMixin
-from polyaxon._flow import V1CompiledOperation, V1Plugins, V1Service
+from polyaxon._flow import V1CompiledOperation, V1Plugins
 
 
 class ServiceConverter(ServiceMixin, BaseConverter):

polyaxon/_flow/environment/__init__.py

@@ -8,6 +8,7 @@ from clipped.compact.pydantic import (
     validation_always,
     validation_before,
 )
+from clipped.types import GenericStr
 
 from polyaxon._k8s import k8s_schemas, k8s_validation
 from polyaxon._schemas.base import BaseSchemaModel
@@ -395,9 +396,9 @@ class V1Environment(BaseSchemaModel):
         "dnsConfig",
     ]
 
-    labels: Optional[Dict[StrictStr, StrictStr]] = None
-    annotations: Optional[Dict[StrictStr, StrictStr]] = None
-    node_selector: Optional[Dict[StrictStr, StrictStr]] = Field(
+    labels: Optional[Dict[StrictStr, GenericStr]] = None
+    annotations: Optional[Dict[StrictStr, GenericStr]] = None
+    node_selector: Optional[Dict[StrictStr, GenericStr]] = Field(
         alias="nodeSelector", default=None
     )
     affinity: Optional[Union[k8s_schemas.V1Affinity, Dict]] = None
@@ -428,7 +429,7 @@ class V1Environment(BaseSchemaModel):
     priority: Optional[int] = None
     restart_policy: Optional[
         Literal["Always", "OnFailure", "Never", "ExitCode"]
-    ] = Field(alias="restartPolicy", default=None)
+    ] = Field(alias="restartPolicy", default=None)  # fmt: skip
 
     @field_validator("affinity", **validation_always, **validation_before)
     def validate_affinity(cls, v):

polyaxon/_flow/io/io.py

@@ -46,6 +46,8 @@ def validate_io_value(
             default=default,
         )
         if validation:
+            if isinstance(validation, dict):
+                validation = V1Validation(**validation)
             validation.run_validation(
                 value=parsed_value, type=type, is_optional=is_optional
             )

polyaxon/_flow/matrix/params.py

@@ -1039,7 +1039,7 @@ class V1HpQLogUniform(BaseHpParamConfig):
     kind: Literal[_IDENTIFIER] = _IDENTIFIER
     value: Optional[
         Union[QLogUniform, QLogUniformList, QLogUniformStr, RefField]
-    ] = None
+    ] = None  # fmt: skip
 
     @property
     def is_distribution(self):

polyaxon/_flow/operations/operation.py

@@ -8,7 +8,6 @@ from clipped.compact.pydantic import (
     field_validator,
     model_validator,
     validation_after,
-    validation_before,
 )
 from clipped.config.patch_strategy import PatchStrategy
 from clipped.config.schema import skip_partial, to_partial

polyaxon/_flow/params/params.py

@@ -386,7 +386,7 @@ class V1Param(BaseSchemaModel, ctx_refs.RefMixin, ParamValueMixin):
 
     _IDENTIFIER = "param"
 
-    value: Any
+    value: Optional[Any] = None
     ref: Optional[StrictStr] = None
     context_only: Optional[bool] = Field(alias="contextOnly", default=False)
     connection: Optional[StrictStr] = None
@@ -479,7 +479,7 @@ class ParamSpec(
         raise PolyaxonValidationError(
             "Param `{}` with type `{}`, "
             "cannot be turned to an init container automatically.".format(
-                self.name, self.type, self.param.ref
+                self.name, self.type
             )
         )
 

polyaxon/_flow/run/dag.py

@@ -444,7 +444,7 @@ class V1Dag(BaseRun):
         for g_context in ctx_sections.GLOBALS_CONTEXTS:
             self._context[
                 "dag.{}.{}".format(ctx_sections.GLOBALS, g_context)
-            ] = V1IO.construct(name=g_context, type="str", value="", is_optional=True)
+            ] = V1IO.construct(name=g_context, type="str", value="", is_optional=True)  # fmt: skip
 
         self._context["dag.{}".format(ctx_sections.INPUTS)] = V1IO.construct(
             name="inputs", type="dict", value={}, is_optional=True
@@ -541,19 +541,19 @@ class V1Dag(BaseRun):
             # We allow to resolve name, status, project, all outputs/inputs, iteration
             self._context[
                 "ops.{}.{}".format(op_name, ctx_sections.INPUTS)
-            ] = V1IO.construct(name="inputs", type="dict", value={}, is_optional=True)
+            ] = V1IO.construct(name="inputs", type="dict", value={}, is_optional=True)  # fmt: skip
             self._context[
                 "ops.{}.{}".format(op_name, ctx_sections.OUTPUTS)
-            ] = V1IO.construct(name="outputs", type="dict", value={}, is_optional=True)
+            ] = V1IO.construct(name="outputs", type="dict", value={}, is_optional=True)  # fmt: skip
             self._context[
                 "ops.{}.{}".format(op_name, ctx_sections.GLOBALS)
-            ] = V1IO.construct(name="globals", type="str", value="", is_optional=True)
+            ] = V1IO.construct(name="globals", type="str", value="", is_optional=True)  # fmt: skip
             self._context[
                 "ops.{}.{}".format(op_name, ctx_sections.ARTIFACTS)
-            ] = V1IO.construct(name="artifacts", type="str", value="", is_optional=True)
+            ] = V1IO.construct(name="artifacts", type="str", value="", is_optional=True)  # fmt: skip
             self._context[
                 "ops.{}.{}".format(op_name, ctx_sections.INPUTS_OUTPUTS)
-            ] = V1IO.construct(name="io", type="str", value={}, is_optional=True)
+            ] = V1IO.construct(name="io", type="str", value={}, is_optional=True)  # fmt: skip
 
         for op in self.operations:
             if op.has_component_reference:

polyaxon/_flow/run/resources.py

@@ -1,4 +1,4 @@
-from typing import ClassVar, Optional, Union, Set
+from typing import ClassVar, Optional, Set, Union
 
 from clipped.compact.pydantic import StrictStr
 from clipped.types.numbers import StrictIntOrFloat

polyaxon/_fs/fs.py

@@ -18,7 +18,7 @@ def _get_fs_from_connection(
     connection: Optional[V1Connection],
     asynchronous: bool = False,
     use_listings_cache: bool = False,
-    **kwargs
+    **kwargs,
 ):
     # We assume that `None` refers to local store as well
     if not connection or connection.kind in {

polyaxon/_k8s/converter/base/containers.py

@@ -73,7 +73,7 @@ class ContainerMixin(BaseConverter):
 
     @staticmethod
     def _sanitize_resources(
-        resources: Union[k8s_schemas.V1ResourceRequirements, Dict]
+        resources: Union[k8s_schemas.V1ResourceRequirements, Dict],
     ) -> Optional[k8s_schemas.V1ResourceRequirements]:
         def validate_resources(r_field: Dict) -> Dict:
             if not r_field:

polyaxon/_k8s/converter/converters/dask_job.py

@@ -20,7 +20,7 @@ class DaskJobConverter(DaskJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1DaskJob
+        job: V1DaskJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1DaskReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/job.py

@@ -18,7 +18,7 @@ class JobConverter(JobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1Job
+        job: V1Job = compiled_operation.run
         plugins = V1Plugins.get_or_create(
             config=compiled_operation.plugins, auth=default_auth
         )

polyaxon/_k8s/converter/converters/kubeflow/mpi_job.py

@@ -20,7 +20,7 @@ class MPIJobConverter(MPIJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1MPIJob
+        job: V1MPIJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/kubeflow/mx_job.py

@@ -20,7 +20,7 @@ class MXJobConverter(MXJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1MXJob
+        job: V1MXJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/kubeflow/paddle_job.py

@@ -20,7 +20,7 @@ class PaddleJobConverter(PaddleJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1PaddleJob
+        job: V1PaddleJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/kubeflow/pytroch_job.py

@@ -20,7 +20,7 @@ class PytorchJobConverter(PytorchJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1PytorchJob
+        job: V1PytorchJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/kubeflow/tf_job.py

@@ -20,7 +20,7 @@ class TfJobConverter(TFJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1TFJob
+        job: V1TFJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/kubeflow/xgboost_job.py

@@ -20,7 +20,7 @@ class XGBoostJobConverter(XGBoostJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1XGBoostJob
+        job: V1XGBoostJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1KFReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/ray_job.py

@@ -1,6 +1,5 @@
 from typing import Dict, Iterable, Optional
 
-from clipped.utils.encoding import encode
 from clipped.utils.json import orjson_dumps
 
 from polyaxon import pkg
@@ -23,7 +22,7 @@ class RayJobConverter(RayJobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> Dict:
-        job = compiled_operation.run  # type: V1RayJob
+        job: V1RayJob = compiled_operation.run
 
         def _get_replica(replica: Optional[V1RayReplica]) -> Optional[ReplicaSpec]:
             if not replica:

polyaxon/_k8s/converter/converters/service.py

@@ -1,7 +1,7 @@
 from typing import Dict, Iterable, Optional
 
 from polyaxon._connections import V1Connection, V1ConnectionResource
-from polyaxon._flow import V1CompiledOperation, V1Plugins, V1Service
+from polyaxon._flow import V1CompiledOperation, V1Plugins
 from polyaxon._k8s.converter.base import BaseConverter
 from polyaxon._k8s.converter.mixins import ServiceMixin
 from polyaxon._k8s.custom_resources.service import get_service_custom_resource

polyaxon/_k8s/k8s_validation.py

@@ -24,13 +24,13 @@ def validate_k8s_affinity(value: Optional[Union[k8s_schemas.V1Affinity, Dict]]):
 
 
 def validate_k8s_security_context(
-    value: Optional[Union[k8s_schemas.V1SecurityContext, Dict]]
+    value: Optional[Union[k8s_schemas.V1SecurityContext, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1SecurityContext)
 
 
 def validate_k8s_pod_dns_config(
-    value: Optional[Union[k8s_schemas.V1PodDNSConfig, Dict]]
+    value: Optional[Union[k8s_schemas.V1PodDNSConfig, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1PodDNSConfig)
 
@@ -56,19 +56,19 @@ def validate_k8s_volume_mount(value: Optional[Union[k8s_schemas.V1VolumeMount, D
 
 
 def validate_k8s_container_port(
-    value: Optional[Union[k8s_schemas.V1ContainerPort, Dict]]
+    value: Optional[Union[k8s_schemas.V1ContainerPort, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1ContainerPort)
 
 
 def validate_k8s_resource_requirements(
-    value: Optional[Union[k8s_schemas.V1ResourceRequirements, Dict]]
+    value: Optional[Union[k8s_schemas.V1ResourceRequirements, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1ResourceRequirements)
 
 
 def validate_k8s_env_from_source(
-    value: Optional[Union[k8s_schemas.V1EnvFromSource, Dict]]
+    value: Optional[Union[k8s_schemas.V1EnvFromSource, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1EnvFromSource)
 
@@ -78,25 +78,25 @@ def validate_k8s_volume(value: Optional[Union[k8s_schemas.V1Volume, Dict]]):
 
 
 def validate_k8s_object_field_selector(
-    value: Optional[Union[k8s_schemas.V1ObjectFieldSelector, Dict]]
+    value: Optional[Union[k8s_schemas.V1ObjectFieldSelector, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1ObjectFieldSelector)
 
 
 def validate_k8s_env_var_source(
-    value: Optional[Union[k8s_schemas.V1EnvVarSource, Dict]]
+    value: Optional[Union[k8s_schemas.V1EnvVarSource, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1EnvVarSource)
 
 
 def validate_k8s_config_map_selector(
-    value: Optional[Union[k8s_schemas.V1ConfigMapKeySelector, Dict]]
+    value: Optional[Union[k8s_schemas.V1ConfigMapKeySelector, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1ConfigMapKeySelector)
 
 
 def validate_k8s_secret_selector(
-    value: Optional[Union[k8s_schemas.V1SecretKeySelector, Dict]]
+    value: Optional[Union[k8s_schemas.V1SecretKeySelector, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1SecretKeySelector)
 
@@ -110,42 +110,42 @@ def validate_k8s_object_meta(value: Optional[Union[k8s_schemas.V1ObjectMeta, Dic
 
 
 def validate_k8s_pod_template_spec(
-    value: Optional[Union[k8s_schemas.V1PodTemplateSpec, Dict]]
+    value: Optional[Union[k8s_schemas.V1PodTemplateSpec, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1PodTemplateSpec)
 
 
 def validate_k8s_host_path_volume_source(
-    value: Optional[Union[k8s_schemas.V1HostPathVolumeSource, Dict]]
+    value: Optional[Union[k8s_schemas.V1HostPathVolumeSource, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1HostPathVolumeSource)
 
 
 def validate_k8s_empty_dir_volume_source(
-    value: Optional[Union[k8s_schemas.V1EmptyDirVolumeSource, Dict]]
+    value: Optional[Union[k8s_schemas.V1EmptyDirVolumeSource, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1EmptyDirVolumeSource)
 
 
 def validate_k8s_persistent_volume_claim_volume_source(
-    value: Optional[Union[k8s_schemas.V1PersistentVolumeClaimVolumeSource, Dict]]
+    value: Optional[Union[k8s_schemas.V1PersistentVolumeClaimVolumeSource, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1PersistentVolumeClaimVolumeSource)
 
 
 def validate_k8s_secret_volume_source(
-    value: Optional[Union[k8s_schemas.V1SecretVolumeSource, Dict]]
+    value: Optional[Union[k8s_schemas.V1SecretVolumeSource, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1SecretVolumeSource)
 
 
 def validate_k8s_config_map_volume_source(
-    value: Optional[Union[k8s_schemas.V1ConfigMapVolumeSource, Dict]]
+    value: Optional[Union[k8s_schemas.V1ConfigMapVolumeSource, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1ConfigMapVolumeSource)
 
 
 def validate_k8s_local_object_reference(
-    value: Optional[Union[k8s_schemas.V1LocalObjectReference, Dict]]
+    value: Optional[Union[k8s_schemas.V1LocalObjectReference, Dict]],
 ):
     return _validate_schema(value, k8s_schemas.V1LocalObjectReference)

polyaxon/_k8s/logging/async_monitor.py

@@ -203,7 +203,7 @@ async def get_op_spec(
     for pod in pods or []:
         pods_list[
             pod.metadata.name
-        ] = k8s_manager.api_client.sanitize_for_serialization(pod)
+        ] = k8s_manager.api_client.sanitize_for_serialization(pod)  # fmt: skip
         pods_list[pod.metadata.name]["events"] = await get_resource_events(
             k8s_manager=k8s_manager,
             resource_type="Pod",
@@ -213,7 +213,7 @@ async def get_op_spec(
     for service in services or []:
         services_list[
             service.metadata.name
-        ] = k8s_manager.api_client.sanitize_for_serialization(service)
+        ] = k8s_manager.api_client.sanitize_for_serialization(service)  # fmt: skip
         services_list[service.metadata.name]["events"] = await get_resource_events(
             k8s_manager=k8s_manager,
             resource_type="Service",
@@ -245,7 +245,7 @@ async def get_agent_spec(
     for pod in pods or []:
         pods_list[
             pod.metadata.name
-        ] = k8s_manager.api_client.sanitize_for_serialization(pod)
+        ] = k8s_manager.api_client.sanitize_for_serialization(pod)  # fmt: skip
         pods_list[pod.metadata.name]["events"] = await get_resource_events(
             k8s_manager=k8s_manager,
             resource_type="Pod",
@@ -256,7 +256,7 @@ async def get_agent_spec(
     for service in services or []:
         services_list[
             service.metadata.name
-        ] = k8s_manager.api_client.sanitize_for_serialization(service)
+        ] = k8s_manager.api_client.sanitize_for_serialization(service)  # fmt: skip
         services_list[service.metadata.name]["events"] = await get_resource_events(
             k8s_manager=k8s_manager,
             resource_type="Service",

polyaxon/_k8s/manager/async_manager.py

@@ -62,7 +62,9 @@ class AsyncK8sManager(BaseK8sManager):
     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]
+        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)
 
     async def _list_namespace_resource(
@@ -108,7 +110,7 @@ class AsyncK8sManager(BaseK8sManager):
         plural,
         reraise: bool = False,
         namespace: str = None,
-        **kwargs
+        **kwargs,
     ) -> List:
         return await self._list_namespace_resource(
             resource_api=self.k8s_custom_object_api.list_namespaced_custom_object,  # type: ignore[attr-defined]
@@ -273,7 +275,7 @@ class AsyncK8sManager(BaseK8sManager):
         field_selector: str = None,
         namespace: str = None,
         reraise: bool = False,
-        **kwargs
+        **kwargs,
     ) -> List[client.CoreV1EventList]:
         return await self._list_namespace_resource(
             resource_api=self.k8s_api.list_namespaced_event,  # type: ignore[attr-defined]