polyaxon 2.6.0rc1__py3-none-any.whl → 2.7.0rc0__py3-none-any.whl

polyaxon/_flow/run/ray/ray.py

@@ -136,6 +136,7 @@ class V1RayJob(BaseRun, DestinationImageMixin):
 
     _IDENTIFIER = V1RunKind.RAYJOB
     _CUSTOM_DUMP_FIELDS = {"head", "workers"}
+    _FIELDS_DICT_PATCH = ["workers"]
 
     kind: Literal[_IDENTIFIER] = _IDENTIFIER
     entrypoint: Optional[str] = None
@@ -196,3 +197,11 @@ class V1RayJob(BaseRun, DestinationImageMixin):
                 worker = self.workers[worker_name]
                 init += worker.get_all_init()
         return init
+
+    def get_replica_types(self):
+        types = []
+        if self.head:
+            types.append("head")
+        if self.workers:
+            types += list(self.workers.keys())
+        return types

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]

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,12 +318,14 @@ 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),
+                validate_run_patch(
+                    preset.run_patch,
+                    config.run.kind,
+                    replica_types=config.get_replica_types(),
+                ),
                 strategy=preset.patch_strategy,
             )
         patch_keys = {
@@ -348,7 +350,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: "
@@ -54,7 +54,11 @@ class OperationSpecification(BaseSpecification):
                 else config.patch_strategy
             )
             component.run = component.run.patch(
-                validate_run_patch(config.run_patch, component.run.kind),
+                validate_run_patch(
+                    config.run_patch,
+                    component.run.kind,
+                    replica_types=component.get_replica_types(),
+                ),
                 strategy=patch_strategy,
             )
 
@@ -107,7 +111,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]