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

polyaxon/_k8s/manager/manager.py

@@ -75,7 +75,7 @@ class K8sManager(BaseK8sManager):
         include_uninitialized=True,
         reraise: bool = False,
         namespace: str = None,
-        **kwargs
+        **kwargs,
     ):
         return self._list_namespace_resource(
             resource_api=self.k8s_api.list_namespaced_pod,
@@ -89,7 +89,7 @@ class K8sManager(BaseK8sManager):
         include_uninitialized=True,
         reraise: bool = False,
         namespace: str = None,
-        **kwargs
+        **kwargs,
     ):
         return self._list_namespace_resource(
             resource_api=self.k8s_batch_api.list_namespaced_job,
@@ -105,7 +105,7 @@ class K8sManager(BaseK8sManager):
         plural,
         reraise: bool = False,
         namespace: str = None,
-        **kwargs
+        **kwargs,
     ):
         return self._list_namespace_resource(
             resource_api=self.k8s_custom_object_api.list_namespaced_custom_object,
@@ -747,7 +747,7 @@ class K8sManager(BaseK8sManager):
         include_uninitialized=True,
         reraise: bool = False,
         namespace: str = None,
-        **kwargs
+        **kwargs,
     ):
         objs = self.list_pods(
             include_uninitialized=include_uninitialized,
@@ -765,7 +765,7 @@ class K8sManager(BaseK8sManager):
         include_uninitialized=True,
         reraise: bool = False,
         namespace: str = None,
-        **kwargs
+        **kwargs,
     ):
         objs = self.list_jobs(
             include_uninitialized=include_uninitialized,
@@ -812,7 +812,7 @@ class K8sManager(BaseK8sManager):
         field_selector: str = None,
         namespace: str = None,
         reraise: bool = False,
-        **kwargs
+        **kwargs,
     ) -> List[client.CoreV1EventList]:
         return self._list_namespace_resource(
             resource_api=self.k8s_api.list_namespaced_event,  # type: ignore[attr-defined]

polyaxon/_local_process/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/_local_process/converter/base/init.py

@@ -50,7 +50,7 @@ class InitConverter(_BaseConverter):
         env_from: List[Any],
         args: List[str],
         command: Optional[List[str]] = None,
-        **kwargs
+        **kwargs,
     ) -> Optional[process_types.V1Container]:
         env = env or []
 

polyaxon/_local_process/converter/base/mounts.py

@@ -1,8 +1,6 @@
 from typing import List, Optional
 
 from polyaxon._connections import V1Connection, V1ConnectionResource
-from polyaxon._contexts import paths as ctx_paths
-from polyaxon._local_process import process_types
 from polyaxon._runner.converter import BaseConverter
 
 

polyaxon/_local_process/converter/converters/job.py

@@ -18,7 +18,7 @@ class JobConverter(JobMixin, BaseConverter):
         default_sa: Optional[str] = None,
         default_auth: bool = False,
     ) -> List[process_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/_local_process/converter/converters/service.py

@@ -1,7 +1,7 @@
 from typing import Dict, Iterable, List, Optional
 
 from polyaxon._connections import V1Connection, V1ConnectionResource
-from polyaxon._flow import V1CompiledOperation, V1Plugins, V1Service
+from polyaxon._flow import V1CompiledOperation, V1Plugins
 from polyaxon._local_process import process_types
 from polyaxon._local_process.converter.base import BaseConverter
 from polyaxon._local_process.converter.mixins import ServiceMixin

polyaxon/_managers/project.py

@@ -22,8 +22,9 @@ class ProjectConfigManager(ConfigManager):
         project = cls.get_config()
         if not project:
             Printer.error(
-                "No project was found, please initialize a project."
-                " {}".format(cli_constants.INIT_COMMAND)
+                "No project was found, please initialize a project." " {}".format(
+                    cli_constants.INIT_COMMAND
+                )
             )
             sys.exit(1)
 

polyaxon/_polyaxonfile/check.py

@@ -93,8 +93,7 @@ def check_polyaxonfile(
 ):
     if sum([1 for i in [python_module, url, hub] if i]) > 1:
         message = (
-            "You can only use one and only one option: "
-            "hub, url, or a python module.".format(hub)
+            "You can only use one and only one option: hub, url, or a python module."
         )
         if is_cli:
             Printer.error(message, sys_exit=True)
@@ -213,8 +212,9 @@ def check_polyaxonfile(
 def check_polyaxonfile_kind(specification, kind):
     if specification.kind != kind:
         Printer.error(
-            "Your polyaxonfile must be of kind: `{}`, "
-            "received: `{}`.".format(kind, specification.kind),
+            "Your polyaxonfile must be of kind: `{}`, " "received: `{}`.".format(
+                kind, specification.kind
+            ),
             sys_exit=True,
         )
 

polyaxon/_polyaxonfile/specs/compiled_operation.py

@@ -87,7 +87,7 @@ class CompiledOperationSpecification(BaseSpecification):
 
     @staticmethod
     def _apply_dag_context(config: V1CompiledOperation) -> V1CompiledOperation:
-        dag_run = config.run  # type: V1Dag
+        dag_run: V1Dag = config.run
         dag_run.process_dag()
         dag_run.validate_dag()
         dag_run.process_components(config.inputs)
@@ -318,9 +318,7 @@ class CompiledOperationSpecification(BaseSpecification):
     ) -> V1CompiledOperation:
         if not preset:
             return config
-        preset = OperationSpecification.read(
-            preset, is_preset=True
-        )  # type: V1Operation
+        preset = OperationSpecification.read(preset, is_preset=True)  # type: V1Operation
         if preset.run_patch:
             config.run = config.run.patch(
                 validate_run_patch(preset.run_patch, config.run.kind),
@@ -348,7 +346,7 @@ class CompiledOperationSpecification(BaseSpecification):
             "conditions",
             "skip_on_upstream_skip",
         }
-        patch_keys = patch_keys.intersection(preset.__fields_set__)
+        patch_keys = patch_keys.intersection(preset.model_fields_set)
         patch_data = {k: getattr(preset, k) for k in patch_keys}
         patch_compiled = V1CompiledOperation.construct(**patch_data)
         return config.patch(patch_compiled, strategy=preset.patch_strategy)

polyaxon/_polyaxonfile/specs/libs/parser.py

@@ -61,9 +61,7 @@ class PolyaxonfileParser:
         return parsed_params
 
     @classmethod
-    def parse_operation(
-        cls, config, param_spec: Dict[str, ParamSpec]
-    ):  # pylint:disable=too-many-branches
+    def parse_operation(cls, config, param_spec: Dict[str, ParamSpec]):  # pylint:disable=too-many-branches
         parsed_params = cls.get_parsed_params(param_spec)
 
         parsed_data = {Sections.VERSION: config.version, Sections.KIND: config.kind}

polyaxon/_polyaxonfile/specs/libs/validator.py

@@ -11,7 +11,7 @@ def validate(spec, data):
 
     def validate_keys(section, config, section_data):
         extra_args = [
-            key for key in section_data.keys() if key not in config.__fields__.keys()
+            key for key in section_data.keys() if key not in config.model_fields.keys()
         ]
         if extra_args:
             raise PolyaxonfileError(

polyaxon/_polyaxonfile/specs/operation.py

@@ -39,7 +39,7 @@ class OperationSpecification(BaseSpecification):
 
             config = config.patch(preset, preset.patch_strategy)
         # Patch run
-        component = config.component  # type: V1Component
+        component: V1Component = config.component
         if not component:
             raise PolyaxonSchemaError(
                 "Compile operation received an invalid configuration: "
@@ -107,7 +107,7 @@ class OperationSpecification(BaseSpecification):
             "conditions",
             "skip_on_upstream_skip",
         }
-        patch_keys = patch_keys.intersection(config.__fields_set__)
+        patch_keys = patch_keys.intersection(config.model_fields_set)
         patch_data = {k: getattr(config, k) for k in patch_keys}
         patch_compiled = V1CompiledOperation.construct(contexts=contexts, **patch_data)
 

polyaxon/_pql/manager.py

@@ -59,7 +59,7 @@ class PQLManager:
             if cls.FIELDS_TRANS[field].get("type"):
                 update_tokenized_query[
                     "{}_type".format(field_trans)
-                ] = cls.FIELDS_TRANS[field]["type"]
+                ] = cls.FIELDS_TRANS[field]["type"]  # fmt: skip
         else:
             update_tokenized_query[key] = tokenized_query[key]
 

polyaxon/_pql/parser.py

@@ -85,8 +85,9 @@ def parse_datetime_operation(operation: str) -> "QueryOpSpec":
     # Check not allowed ops
     if "|" in _operation:
         raise PQLException(
-            "`|` is not allowed for datetime conditions. "
-            "Operation: {}".format(operation)
+            "`|` is not allowed for datetime conditions. Operation: {}".format(
+                operation
+            )
         )
 
     # Check nil operation
@@ -140,13 +141,11 @@ def parse_scalar_operation(operation: str, loader: Callable = None) -> "QueryOpS
     # Check not allowed ops
     if "|" in _operation:
         raise PQLException(
-            "`|` is not allowed for scalar conditions. "
-            "Operation: {}".format(operation)
+            "`|` is not allowed for scalar conditions. Operation: {}".format(operation)
         )
     if ".." in _operation:
         raise PQLException(
-            "`..` is not allowed for scalar conditions. "
-            "Operation: {}".format(operation)
+            "`..` is not allowed for scalar conditions. Operation: {}".format(operation)
         )
 
     # Check nil operation
@@ -173,8 +172,9 @@ def parse_scalar_operation(operation: str, loader: Callable = None) -> "QueryOpS
             float(_operation)
         except (ValueError, TypeError):
             raise PQLException(
-                "Scalar operation requires int or float params, "
-                "received {}.".format(operation)
+                "Scalar operation requires int or float params, received {}.".format(
+                    operation
+                )
             )
     return QueryOpSpec(op, negation, _operation)
 
@@ -205,8 +205,7 @@ def parse_value_operation(operation: str) -> "QueryOpSpec":
     # Check range not allowed
     if ".." in _operation:
         raise PQLException(
-            "`..` is not allowed for value conditions. "
-            "Operation: {}".format(operation)
+            "`..` is not allowed for value conditions. Operation: {}".format(operation)
         )
 
     # Check nil operation
@@ -230,8 +229,9 @@ def parse_value_operation(operation: str) -> "QueryOpSpec":
         ]
         if len(params) <= 1:
             raise PQLException(
-                "`{}` is not allowed for value conditions, "
-                "Operation: {}".format(op, operation)
+                "`{}` is not allowed for value conditions, Operation: {}".format(
+                    op, operation
+                )
             )
         return QueryOpSpec(op, negation, params)
 
@@ -270,8 +270,7 @@ def parse_search_operation(operation: str) -> "QueryOpSpec":
     # Check range not allowed
     if ".." in _operation:
         raise PQLException(
-            "`..` is not allowed for search conditions. "
-            "Operation: {}".format(operation)
+            "`..` is not allowed for search conditions. Operation: {}".format(operation)
         )
 
     # Check nil operation
@@ -286,8 +285,9 @@ def parse_search_operation(operation: str) -> "QueryOpSpec":
     op, _operation = parse_comparison_operation(_operation)
     if op:
         raise PQLException(
-            "`{}` is not allowed for search conditions, "
-            "Operation: {}".format(op, operation)
+            "`{}` is not allowed for search conditions, Operation: {}".format(
+                op, operation
+            )
         )
 
     # Early return
@@ -303,8 +303,9 @@ def parse_search_operation(operation: str) -> "QueryOpSpec":
         ]
         if len(params) <= 1:
             raise PQLException(
-                "`{}` is not allowed for search conditions, "
-                "Operation: {}".format(op, operation)
+                "`{}` is not allowed for search conditions, Operation: {}".format(
+                    op, operation
+                )
             )
         return QueryOpSpec(op, negation, params)
 
@@ -317,8 +318,9 @@ def parse_search_operation(operation: str) -> "QueryOpSpec":
         params = [param.strip() for param in params if param.strip()]
         if len(params) != 1:
             raise PQLException(
-                "`{}` is not allowed for search conditions, "
-                "Operation: {}".format(op, operation)
+                "`{}` is not allowed for search conditions, Operation: {}".format(
+                    op, operation
+                )
             )
         return QueryOpSpec(op, negation, parse_operation_value(params[0]))
 
@@ -328,8 +330,9 @@ def parse_search_operation(operation: str) -> "QueryOpSpec":
         params = [param.strip() for param in params if param.strip()]
         if len(params) != 1:
             raise PQLException(
-                "`{}` is not allowed for search conditions, "
-                "Operation: {}".format(op, operation)
+                "`{}` is not allowed for search conditions, Operation: {}".format(
+                    op, operation
+                )
             )
         return QueryOpSpec(op, negation, parse_operation_value(params[0]))
 
@@ -339,16 +342,16 @@ def parse_search_operation(operation: str) -> "QueryOpSpec":
         params = [param.strip() for param in params if param.strip()]
         if len(params) != 1:
             raise PQLException(
-                "`{}` is not allowed for search conditions, "
-                "Operation: {}".format(op, operation)
+                "`{}` is not allowed for search conditions, Operation: {}".format(
+                    op, operation
+                )
             )
         return QueryOpSpec(op, negation, parse_operation_value(params[0]))
 
     if not _operation:
         raise PQLException(
             "Expression is not valid, it must be formatted as "
-            "name:operation, "
-            "Operation: {}".format(operation)
+            "name:operation, Operation: {}".format(operation)
         )
     # Now the operation must be an equality param param
     return QueryOpSpec("=", negation, _operation)

polyaxon/_runner/agent/base_agent.py

@@ -70,8 +70,7 @@ class BaseAgent:
             )
         except Exception as e:
             logger.warning(
-                "Agent failed to collect agent data: {}\n"
-                "Retrying ...".format(repr(e))
+                "Agent failed to collect agent data: {}\nRetrying ...".format(repr(e))
             )
 
     def sync_compatible_updates(self, compatible_updates: Dict):

polyaxon/_runner/converter/init/artifacts.py

@@ -8,8 +8,9 @@ def get_artifacts_store_args(artifacts_path: str, clean: bool) -> str:
         dir=artifacts_path
     )
     delete_dir = (
-        'if [ -d {path} ] && [ "$(ls -A {path})" ]; '
-        "then rm -R {path}/*; fi;".format(path=artifacts_path)
+        'if [ -d {path} ] && [ "$(ls -A {path})" ]; then rm -R {path}/*; fi;'.format(
+            path=artifacts_path
+        )
     )
     if clean:
         return "{} {}".format(get_or_create, delete_dir)

polyaxon/_runner/converter/init/tensorboard.py

@@ -21,7 +21,7 @@ def get_tensorboard_args(
     if tb_args.use_names:
         args.append("--use-names")
     if tb_args.path_prefix:
-        args.append("--path-prefix={}".format(tb_args.path_prefix)),
+        args.append("--path-prefix={}".format(tb_args.path_prefix))
     if tb_args.plugins:
         plugins = validate_tags(tb_args.plugins, validate_yaml=True)
         args.append("--plugins={}".format(",".join(plugins)))

polyaxon/_schemas/client.py

@@ -113,7 +113,7 @@ class ClientConfig(BaseSchemaModel):
         token: Optional[str] = None,
         use_cloud_host: bool = False,
         retries: Optional[int] = None,
-        **data
+        **data,
     ):
         host = (
             clean_host(host or LOCALHOST) if not use_cloud_host else POLYAXON_CLOUD_HOST