polyaxon 2.0.0rc49__py3-none-any.whl → 2.4.0rc1__py3-none-any.whl

polyaxon/_deploy/schemas/deployment.py

@@ -175,7 +175,7 @@ def validate_deployment_chart(
     if deployment_chart == DeploymentCharts.AGENT and not agent:
         raise ValueError("Agent deployment requires a valid `agent` key configuration.")
 
-    if deployment_chart == DeploymentCharts.PLATFORM and agent:
+    if deployment_chart == DeploymentCharts.PLATFORM and agent and agent.enabled:
         Printer.header("Deployment has agent enabled!")
 
 
@@ -198,6 +198,7 @@ class DeploymentConfig(BaseSchemaModel):
     polyaxon_secret: Optional[StrictStr] = Field(alias="polyaxonSecret")
     internal_token: Optional[StrictStr] = Field(alias="internalToken")
     password_length: Optional[StrictInt] = Field(alias="passwordLength")
+    password_auth: Optional[bool] = Field(alias="passwordAuth")
     ssl: Optional[SSLConfig]
     encryption_secret: Optional[StrictStr] = Field(alias="encryptionSecret")
     platform_secret: Optional[StrictStr] = Field(alias="platformSecret")
@@ -254,6 +255,7 @@ class DeploymentConfig(BaseSchemaModel):
     allowed_hosts: Optional[List[StrictStr]] = Field(alias="allowedHosts")
     include_host_ips: Optional[bool] = Field(alias="includeHostIps")
     intervals: Optional[IntervalsConfig]
+    cleaning_intervals: Optional[Dict] = Field(alias="cleaningIntervals")
     artifacts_store: Optional[V1Connection] = Field(alias="artifactsStore")
     connections: Optional[List[V1Connection]]
     mount_connections: Optional[List[str]] = Field(alias="mountConnections")
@@ -293,6 +295,7 @@ class DeploymentConfig(BaseSchemaModel):
         if values.get("deployment_chart") == DeploymentCharts.AGENT:
             wrong_agent_deployment_keys(
                 password_length=values.get("password_length"),
+                password_auth=values.get("password_auth"),
                 platform_secret=values.get("platform_secret"),
                 encryption_secret=values.get("encryption_secret"),
                 user=values.get("user"),

polyaxon/_deploy/schemas/intervals.py

@@ -6,11 +6,4 @@ from polyaxon._schemas.base import BaseSchemaModel
 
 
 class IntervalsConfig(BaseSchemaModel):
-    runs_scheduler: Optional[StrictInt] = Field(alias="runsScheduler")
-    operations_default_retry_delay: Optional[StrictInt] = Field(
-        alias="operationsDefaultRetryDelay"
-    )
-    operations_max_retry_delay: Optional[StrictInt] = Field(
-        alias="operationsMaxRetryDelay"
-    )
     compatibility_check: Optional[StrictInt] = Field(alias="compatibilityCheck")

polyaxon/_deploy/schemas/proxy.py

@@ -13,4 +13,5 @@ class ProxyConfig(BaseSchemaModel):
     no_proxy: Optional[StrictStr] = Field(alias="noProxy")
     port: Optional[StrictInt]
     host: Optional[StrictStr]
+    protocol: Optional[StrictStr]
     kind: Optional[StrictStr]

polyaxon/_deploy/schemas/service.py

@@ -12,6 +12,7 @@ class DeploymentService(BaseServiceConfig):
     enabled: Optional[bool]
     replicas: Optional[StrictInt]
     concurrency: Optional[StrictInt]
+    per_core: Optional[bool] = Field(alias="perCore")
     scheme: Optional[StrictStr]
 
 
@@ -22,6 +23,11 @@ class WorkerServiceConfig(DeploymentService):
 class AgentServiceConfig(DeploymentService):
     instance: Optional[StrictStr]
     token: Optional[StrictStr]
+    watch_cluster: Optional[bool] = Field(alias="watchCluster")
+    additional_namespaces: Optional[List[str]] = Field(alias="additionalNamespaces")
+    service_account_annotations: Optional[Dict] = Field(
+        alias="serviceAccountAnnotations"
+    )
     is_replica: Optional[bool] = Field(alias="isReplica")
 
 
@@ -50,6 +56,7 @@ class HooksConfig(DeploymentService):
     tables: Optional[bool] = Field(alias="tables")
     sync_db: Optional[bool] = Field(alias="syncdb")
     admin_user: Optional[bool] = Field(alias="adminUser")
+    default_org: Optional[bool] = Field(alias="defaultOrg")
 
 
 class ThirdPartyService(DeploymentService):
@@ -92,6 +99,7 @@ class ExternalService(BaseSchemaModel):
 class ExternalBackend(BaseSchemaModel):
     enabled: Optional[bool]
     backend: Optional[StrictStr]
+    url: Optional[StrictStr]
     options: Optional[Dict]
 
 
@@ -100,7 +108,9 @@ class AuthServicesConfig(BaseSchemaModel):
     gitlab: Optional[ExternalBackend]
     bitbucket: Optional[ExternalBackend]
     google: Optional[ExternalBackend]
-    saml: Optional[ExternalBackend]
+    okta: Optional[ExternalBackend]
+    onelogin: Optional[ExternalBackend]
+    azuread: Optional[ExternalBackend]
 
 
 class ExternalServicesConfig(BaseSchemaModel):

polyaxon/_docker/converter/base/base.py

@@ -123,6 +123,14 @@ class BaseConverter(
         if not resources:
             return None
 
+        if isinstance(resources, dict):
+            try:
+                resources = k8s_schemas.V1ResourceRequirements(**resources)
+            except (ValueError, TypeError) as e:
+                raise PolyaxonConverterError(
+                    f"Could not parse resources value `{resources}` from the K8S schema in container section"
+                ) from e
+
         cpus = None
         memory = None
         gpus = None

polyaxon/_docker/executor.py

@@ -35,6 +35,7 @@ class Executor(BaseExecutor):
         run_uuid: str,
         run_kind: str,
         resource: List[docker_types.V1Container],
+        namespace: str = None,
     ) -> Dict:
         logger.info(f"[Executor] Starting operation {run_uuid} {run_kind}.")
         self._ops[run_uuid] = []
@@ -62,13 +63,15 @@ class Executor(BaseExecutor):
         self._clean_temp_execution_path(run_uuid)
         return {"status": V1Statuses.SUCCEEDED, "tasks": self._ops[run_uuid]}
 
-    def apply(self, run_uuid: str, run_kind: str, resource: Dict) -> Dict:
+    def apply(
+        self, run_uuid: str, run_kind: str, resource: Dict, namespace: str = None
+    ) -> Dict:
         raise PolyaxonAgentError(
             "Docker executor does not support apply method.\n"
             "Run: <kind: {}, uuid: {}>".format(run_kind, run_uuid)
         )
 
-    def stop(self, run_uuid: str, run_kind: str):
+    def stop(self, run_uuid: str, run_kind: str, namespace: str = None):
         proc = self._get_op_proc(run_uuid)
         if proc.poll() is None:
             # Kill the process tree rooted at the child if it's the leader of its own process
@@ -85,7 +88,7 @@ class Executor(BaseExecutor):
                 logger.debug(_msg)
             proc.wait()
 
-    def clean(self, run_uuid: str, run_kind: str):
+    def clean(self, run_uuid: str, run_kind: str, namespace: str = None):
         return self.apply(
             run_uuid=run_uuid,
             run_kind=run_kind,
@@ -100,6 +103,9 @@ class Executor(BaseExecutor):
             return V1Statuses.SUCCEEDED
         return V1Statuses.FAILED
 
-    def get(self, run_uuid: str, run_kind: str) -> V1Statuses:
+    def get(self, run_uuid: str, run_kind: str, namespace: str = None) -> V1Statuses:
         procs = self._get_op_proc(run_uuid)
         return self._get_task_status(procs[-1])
+
+    def list_ops(self, namespace: str = None):
+        return []

polyaxon/_env_vars/getters/owner_entity.py

@@ -5,7 +5,7 @@ from clipped.utils.strings import validate_slug
 
 from polyaxon._constants.globals import DEFAULT
 from polyaxon._env_vars.getters.user import get_local_owner
-from polyaxon._utils.fqn_utils import get_entity_info
+from polyaxon._utils.fqn_utils import get_entity_info, split_owner_team_space
 from polyaxon.exceptions import PolyaxonClientException, PolyaxonSchemaError
 
 
@@ -31,6 +31,8 @@ def resolve_entity_info(entity: str, entity_name: str, is_cli: bool = False):
     if not owner:
         owner = settings.AUTH_CONFIG.username if settings.AUTH_CONFIG else None
 
+    owner, team = split_owner_team_space(owner)
+
     if not all([owner, entity_value]):
         message = "Please provide a valid {}.".format(entity_name)
         if is_cli:
@@ -49,4 +51,4 @@ def resolve_entity_info(entity: str, entity_name: str, is_cli: bool = False):
                 entity_name, entity_value
             )
         )
-    return owner, entity_value
+    return owner, team, entity_value

polyaxon/_env_vars/getters/project.py

@@ -7,7 +7,7 @@ from polyaxon._constants.globals import DEFAULT
 from polyaxon._env_vars.getters.user import get_local_owner
 from polyaxon._managers.project import ProjectConfigManager
 from polyaxon._utils.cache import get_local_project
-from polyaxon._utils.fqn_utils import get_entity_info
+from polyaxon._utils.fqn_utils import get_entity_info, split_owner_team_space
 from polyaxon.exceptions import PolyaxonClientException, PolyaxonSchemaError
 
 
@@ -52,6 +52,8 @@ def get_project_or_local(project=None, is_cli: bool = False):
     if not owner and (not settings.CLI_CONFIG or settings.CLI_CONFIG.is_community):
         owner = DEFAULT
 
+    owner, team = split_owner_team_space(owner)
+
     if not all([owner, project_name]):
         error_message = get_project_error_message(owner, project_name)
         if is_cli:
@@ -79,4 +81,4 @@ def get_project_or_local(project=None, is_cli: bool = False):
             sys.exit(1)
         else:
             raise PolyaxonSchemaError(error_message)
-    return owner, project_name
+    return owner, team, project_name

polyaxon/_env_vars/getters/run.py

@@ -37,9 +37,9 @@ def get_run_or_local(run_uuid=None, is_cli: bool = False):
 
 
 def get_project_run_or_local(project=None, run_uuid=None, is_cli: bool = True):
-    user, project_name = get_project_or_local(project, is_cli=is_cli)
+    owner, team, project_name = get_project_or_local(project, is_cli=is_cli)
     run_uuid = get_run_or_local(run_uuid, is_cli=is_cli)
-    return user, project_name, run_uuid
+    return owner, team, project_name, run_uuid
 
 
 def get_collect_artifacts(arg: Optional[bool] = None, default: Optional[bool] = None):
@@ -74,7 +74,10 @@ def get_run_info(run_instance: Optional[str] = None):
             "Could not get run info, "
             "please make sure this is run is correctly started by Polyaxon."
         )
+    return get_run_info_from_instance(run_instance)
 
+
+def get_run_info_from_instance(run_instance: str):
     parts = run_instance.split(".")
     if not len(parts) == 4:
         raise PolyaxonClientException(

polyaxon/_env_vars/keys.py

@@ -78,6 +78,9 @@ ENV_KEYS_AGENT_ARTIFACTS_STORE = "POLYAXON_AGENT_ARTIFACTS_STORE"
 ENV_KEYS_AGENT_CONNECTIONS = "POLYAXON_AGENT_CONNECTIONS"
 ENV_KEYS_SET_AGENT = "POLYAXON_SET_AGENT"
 ENV_KEYS_K8S_APP_SECRET_NAME = "POLYAXON_K8S_APP_SECRET_NAME"  # noqa
+ENV_KEYS_WATCH_CLUSTER = "POLYAXON_WATCH_CLUSTER"
+ENV_KEYS_SINGLE_NAMESPACE = "POLYAXON_SINGLE_NAMESPACE"
+ENV_KEYS_ADDITIONAL_NAMESPACES = "POLYAXON_ADDITIONAL_NAMESPACES"
 ENV_KEYS_AGENT_SECRET_NAME = "POLYAXON_AGENT_SECRET_NAME"  # noqa
 ENV_KEYS_AGENT_RUNS_SA = "POLYAXON_AGENT_RUNS_SA"
 ENV_KEYS_AGENT_ENABLE_HEALTH_CHECKS = "POLYAXON_AGENT_ENABLE_HEALTH_CHECKS"
@@ -112,14 +115,16 @@ ENV_KEYS_SANDBOX_DEBUG = "POLYAXON_SANDBOX_DEBUG"
 ENV_KEYS_SANDBOX_SSL_ENABLED = "POLYAXON_SANDBOX_SSL_ENABLED"
 ENV_KEYS_SANDBOX_WORKERS = "POLYAXON_SANDBOX_WORKERS"
 ENV_KEYS_SANDBOX_PER_CORE = "POLYAXON_SANDBOX_PER_CORE"
-ENV_KEYS_SANDBOX_IS_LOCAL = "POLYAXON_SANDBOX_IS_LOCAL"
 ENV_KEYS_SANDBOX_ROOT = "POLYAXON_SANDBOX_ROOT"
+ENV_KEYS_SERVICE_MODE = "POLYAXON_SERVICE_MODE"
 
 # Proxies
 ENV_KEYS_PROXY_NAMESPACES = "POLYAXON_PROXY_NAMESPACES"
 ENV_KEYS_PROXY_GATEWAY_PORT = "POLYAXON_PROXY_GATEWAY_PORT"
 ENV_KEYS_PROXY_GATEWAY_TARGET_PORT = "POLYAXON_PROXY_GATEWAY_TARGET_PORT"
 ENV_KEYS_PROXY_GATEWAY_HOST = "POLYAXON_PROXY_GATEWAY_HOST"
+ENV_KEYS_PROXY_GATEWAY_CONCURRENCY = "POLYAXON_PROXY_GATEWAY_CONCURRENCY"
+ENV_KEYS_PROXY_GATEWAY_PER_CORE = "POLYAXON_PROXY_GATEWAY_PER_CORE"
 ENV_KEYS_PROXY_STREAMS_PORT = "POLYAXON_PROXY_STREAMS_PORT"
 ENV_KEYS_PROXY_STREAMS_TARGET_PORT = "POLYAXON_PROXY_STREAMS_TARGET_PORT"
 ENV_KEYS_PROXY_STREAMS_HOST = "POLYAXON_PROXY_STREAMS_HOST"
@@ -136,6 +141,7 @@ ENV_KEYS_PROXY_AUTH_USE_RESOLVER = "POLYAXON_PROXY_AUTH_USE_RESOLVER"
 ENV_KEYS_PROXY_HAS_FORWARD_PROXY = "POLYAXON_PROXY_HAS_FORWARD_PROXY"
 ENV_KEYS_PROXY_FORWARD_PROXY_PORT = "POLYAXON_PROXY_FORWARD_PROXY_PORT"
 ENV_KEYS_PROXY_FORWARD_PROXY_HOST = "POLYAXON_PROXY_FORWARD_PROXY_HOST"
+ENV_KEYS_PROXY_FORWARD_PROXY_PROTOCOL = "POLYAXON_PROXY_FORWARD_PROXY_PROTOCOL"
 ENV_KEYS_PROXY_FORWARD_PROXY_KIND = "POLYAXON_PROXY_FORWARD_PROXY_KIND"
 ENV_KEYS_UI_IN_SANDBOX = "POLYAXON_UI_IN_SANDBOX"
 ENV_KEYS_UI_ADMIN_ENABLED = "POLYAXON_UI_ADMIN_ENABLED"

polyaxon/_flow/__init__.py

@@ -75,8 +75,10 @@ from polyaxon._flow.run import (
     V1MPIJob,
     V1MXJob,
     V1NotifierJob,
+    V1PaddleElasticPolicy,
     V1PaddleJob,
     V1PipelineKind,
+    V1PytorchElasticPolicy,
     V1PytorchJob,
     V1RayJob,
     V1RayReplica,

polyaxon/_flow/builds/__init__.py

@@ -28,6 +28,7 @@ class V1Build(BaseSchemaModel):
         hub_ref: str
         connection: str
         queue: str, optional
+        namespace: str, optional
         presets: List[str], optional
         cache: [V1Cache](/docs/automation/helpers/cache/), optional
         params: Dict[str, [V1Param](/docs/core/specification/params/)], optional
@@ -94,6 +95,17 @@ class V1Build(BaseSchemaModel):
     ...
     ```
 
+    ### presets
+
+    The [presets](/docs/management/organizations/presets/) to use for the hook operation,
+    if provided, it will override the component's presets otherwise
+    the presets of the component will be used if available.
+
+    ```yaml
+    >>> build:
+    >>>   presets: [test]
+    ```
+
     ### queue
 
     The [queue](/docs/core/scheduling-strategies/queues/) to use.
@@ -112,15 +124,15 @@ class V1Build(BaseSchemaModel):
     >>>   queue: queue-name
     ```
 
-    ### presets
+    ### namespace
 
-    The [presets](/docs/management/organizations/presets/) to use for the hook operation,
-    if provided, it will override the component's presets otherwise
-    the presets of the component will be used if available.
+    > **Note**: Please note that this field is only available in some commercial editions.
+
+    The namespace to use, if not provided, it will default to the agent's namespace.
 
     ```yaml
     >>> build:
-    >>>   presets: [test]
+    >>>   namespace: polyaxon
     ```
 
     ### cache
@@ -211,8 +223,9 @@ class V1Build(BaseSchemaModel):
     _IDENTIFIER = "build"
     hub_ref: StrictStr = Field(alias="hubRef")
     connection: Optional[StrictStr]
-    queue: Optional[StrictStr]
     presets: Optional[Union[List[StrictStr], RefField]]
+    queue: Optional[StrictStr]
+    namespace: Optional[StrictStr]
     cache: Optional[Union[V1Cache, RefField]]
     params: Optional[Dict[str, Union[V1Param, RefField]]]
     run_patch: Optional[Dict[str, Any]] = Field(alias="runPatch")

polyaxon/_flow/component/base.py

@@ -20,6 +20,7 @@ class BaseComponent(BaseSchemaModel):
     tags: Optional[List[StrictStr]]
     presets: Optional[List[StrictStr]]
     queue: Optional[StrictStr]
+    namespace: Optional[StrictStr]
     cache: Optional[Union[V1Cache, RefField]]
     termination: Optional[Union[V1Termination, RefField]]
     plugins: Optional[Union[V1Plugins, RefField]]

polyaxon/_flow/component/component.py

@@ -39,6 +39,7 @@ class V1Component(
         tags: List[str], optional
         presets: List[str], optional
         queue: str, optional
+        namespace: str, optional
         cache: [V1Cache](/docs/automation/helpers/cache/), optional
         termination: [V1Termination](/docs/core/specification/termination/), optional
         plugins: [V1Plugins](/docs/core/specification/plugins/), optional
@@ -60,6 +61,7 @@ class V1Component(
     >>>   tags:
     >>>   presets:
     >>>   queue:
+    >>>   namespace:
     >>>   cache:
     >>>   termination:
     >>>   plugins:
@@ -85,6 +87,7 @@ class V1Component(
     >>>     tags=["test"],
     >>>     presets=["test"],
     >>>     queue="test",
+    >>>     namespace="test",
     >>>     cache=V1Cache(...),
     >>>     termination=V1Termination(...),
     >>>     plugins=V1Plugins(...),
@@ -197,6 +200,17 @@ class V1Component(
     >>>   queue: queue-name
     ```
 
+    ### namespace
+
+    > **Note**: Please note that this field is only available in some commercial editions.
+
+    The namespace to use, if not provided, it will default to the agent's namespace.
+
+    ```yaml
+    >>> component:
+    >>>   namespace: polyaxon
+    ```
+
     ### cache
 
     The default component [cache](/docs/automation/helpers/cache/).

polyaxon/_flow/environment/__init__.py

@@ -207,14 +207,14 @@ class V1Environment(BaseSchemaModel):
     ```yaml
     >>> rules:
     >>>   - apiGroups: [""]
-    >>>     resources: ["pods"]
+    >>>     resources: ["pods", "services", "events", "pods/status", "pods/log"]
     >>>     verbs: ["get", "watch", "list"]
     >>>   - apiGroups: ["metrics.k8s.io"]
-    >>>     resources: ["pods", "nodes", "apis"]
+    >>>     resources: ["pods", "nodes"]
     >>>     verbs: ["get", "list", "watch"]
-    >>>   - apiGroups: ["", "*"]
-    >>>     resources: ["events", "pods/status", "pods/log"]
-    >>>     verbs: ["watch", "get", "list"]
+    >>>   - apiGroups: ["core.polyaxon.com"]
+    >>>     resources: ["operations"]
+    >>>     verbs: ["get", "watch", "list"]
     ```
 
     ### hostAliases
@@ -408,9 +408,9 @@ class V1Environment(BaseSchemaModel):
     scheduler_name: Optional[StrictStr] = Field(alias="schedulerName")
     priority_class_name: Optional[StrictStr] = Field(alias="priorityClassName")
     priority: Optional[int]
-    restart_policy: Optional[Literal["Always", "OnFailure", "Never"]] = Field(
-        alias="restartPolicy"
-    )
+    restart_policy: Optional[
+        Literal["Always", "OnFailure", "Never", "ExitCode"]
+    ] = Field(alias="restartPolicy")
 
     @validator("affinity", always=True, pre=True)
     def validate_affinity(cls, v):

polyaxon/_flow/hooks/__init__.py

@@ -49,6 +49,7 @@ class V1Hook(BaseSchemaModel):
         hub_ref: str
         conditions: str, optional
         queue: str, optional
+        namespace: str, optional
         presets: List[str], optional
         disableDefaults: bool, optional
         params: Dict[str, [V1Param](/docs/core/specification/params/)], optional
@@ -131,6 +132,17 @@ class V1Hook(BaseSchemaModel):
     In the example above, the hook will only run if a param is passed, or an output is logged and
     is equal to "some-value".
 
+    ### presets
+
+    The [presets](/docs/management/organizations/presets/) to use for the hook operation,
+    if provided, it will override the component's presets otherwise
+    the presets of the component will be used if available.
+
+    ```yaml
+    >>> hook:
+    >>>   presets: [test]
+    ```
+
     ### queue
 
     The [queue](/docs/core/scheduling-strategies/queues/) to use.
@@ -149,15 +161,15 @@ class V1Hook(BaseSchemaModel):
     >>>   queue: queue-name
     ```
 
-    ### presets
+    ### namespace
 
-    The [presets](/docs/management/organizations/presets/) to use for the hook operation,
-    if provided, it will override the component's presets otherwise
-    the presets of the component will be used if available.
+    > **Note**: Please note that this field is only available in some commercial editions.
+
+    The namespace to use, if not provided, it will default to the agent's namespace.
 
     ```yaml
     >>> hook:
-    >>>   presets: [test]
+    >>>   namespace: polyaxon
     ```
 
     ### disableDefaults
@@ -199,8 +211,9 @@ class V1Hook(BaseSchemaModel):
     connection: Optional[StrictStr]
     trigger: Optional[V1Statuses]
     conditions: Optional[StrictStr]
-    queue: Optional[StrictStr]
     presets: Optional[List[StrictStr]]
+    queue: Optional[StrictStr]
+    namespace: Optional[StrictStr]
     params: Optional[Union[Dict[str, V1Param], RefField]]
     disable_defaults: Optional[BoolOrRef] = Field(alias="disableDefaults")
 

polyaxon/_flow/init/__init__.py

@@ -106,8 +106,8 @@ class V1Init(BaseSchemaModel):
 
     ```python
     >>> from polyaxon.schemas import V1Component, V1Init, V1Job
-    >>> from polyaxon.schemas.types import V1ArtifactsType, V1DockerfileType, V1GitType
-    >>> from polyaxon.k8s import k8s_schemas
+    >>> from polyaxon.types import V1ArtifactsType, V1DockerfileType, V1GitType
+    >>> from polyaxon import k8s
     >>> component = V1Component(
     >>>     run=V1Job(
     >>>        init=[
@@ -132,8 +132,8 @@ class V1Init(BaseSchemaModel):
     >>>             V1Init(
     >>>                 connection="gcs-large-datasets",
     >>>                 artifacts=V1ArtifactsType(dirs=["data"]),
-    >>>                 container=k8s_schemas.V1Container(
-    >>>                     resources=k8s_schemas.V1ResourceRequirements(requests={"memory": "256Mi", "cpu": "500m"}), # noqa
+    >>>                 container=k8s.V1Container(
+    >>>                     resources=k8s.V1ResourceRequirements(requests={"memory": "256Mi", "cpu": "500m"}), # noqa
     >>>                 )
     >>>             ),
     >>>             V1Init(
@@ -163,14 +163,14 @@ class V1Init(BaseSchemaModel):
     >>>                 )
     >>>             ),
     >>>             V1Init(
-    >>>                 container=k8s_schemas.V1Container(
+    >>>                 container=k8s.V1Container(
     >>>                     name="myapp-container",
     >>>                     image="busybox:1.28",
     >>>                     command=['sh', '-c', 'echo custom init container']
     >>>                 )
     >>>             ),
     >>>        ],
-    >>>        container=k8s_schemas.V1Container(...)
+    >>>        container=k8s.V1Container(...)
     >>>     )
     >>> )
     ```

polyaxon/_flow/matrix/iterative.py

@@ -43,7 +43,6 @@ class V1Iterative(BaseSearchConfig):
     ## Python usage
 
     ```python
-    >>> from polyaxon.k8s import k8s_schemas
     >>> from polyaxon.schemas import (
     >>>     V1Iterative,
     >>>     V1HpLogSpace,

polyaxon/_flow/matrix/tuner.py

@@ -60,6 +60,17 @@ class V1Tuner(BaseSchemaModel):
     ...
     ```
 
+    ### presets
+
+    The [presets](/docs/management/organizations/presets/) to use for the tuner operation,
+    if provided, it will override the component's presets otherwise
+    the presets of the component will be used if available.
+
+    ```yaml
+    >>> tuner:
+    >>>   presets: [test]
+    ```
+
     ### queue
 
     The [queue](/docs/core/scheduling-strategies/queues/) to use.
@@ -78,15 +89,15 @@ class V1Tuner(BaseSchemaModel):
     >>>   queue: queue-name
     ```
 
-    ### presets
+    ### namespace
 
-    The [presets](/docs/management/organizations/presets/) to use for the tuner operation,
-    if provided, it will override the component's presets otherwise
-    the presets of the component will be used if available.
+    > **Note**: Please note that this field is only available in some commercial editions.
+
+    The namespace to use, if not provided, it will default to the agent's namespace.
 
     ```yaml
     >>> tuner:
-    >>>   presets: [test]
+    >>>   namespace: polyaxon
     ```
 
     ### params
@@ -108,6 +119,7 @@ class V1Tuner(BaseSchemaModel):
     _IDENTIFIER = "tuner"
 
     hub_ref: StrictStr = Field(alias="hubRef")
-    queue: Optional[StrictStr]
     presets: Optional[Union[List[StrictStr], RefField]]
+    queue: Optional[StrictStr]
+    namespace: Optional[StrictStr]
     params: Optional[Union[Dict[str, V1Param], RefField]]