polyaxon 2.2.0rc0__py3-none-any.whl → 2.2.0rc1__py3-none-any.whl

polyaxon/_cli/operations.py

@@ -21,7 +21,11 @@ from clipped.utils.versions import compare_versions
 from urllib3.exceptions import HTTPError
 
 from polyaxon import settings
-from polyaxon._cli.dashboard import get_dashboard, get_dashboard_url
+from polyaxon._cli.dashboard import (
+    get_dashboard,
+    get_dashboard_url,
+    get_project_subpath_url,
+)
 from polyaxon._cli.errors import handle_cli_error, is_in_ce
 from polyaxon._cli.options import (
     OPTIONS_NAME,
@@ -286,7 +290,7 @@ def ls(
             else:
                 Printer.warning(f"Skipping run {uid}, offline data not found.")
     else:
-        owner, project_name = get_project_or_local(
+        owner, _, project_name = get_project_or_local(
             project or ctx.obj.get("project"), is_cli=True
         )
 
@@ -431,7 +435,7 @@ def get(ctx, project, uid, offline, path, output):
             sys.exit(1)
         run_data = RunConfigManager.read_from_path(offline_path)
     else:
-        owner, project_name, run_uuid = get_project_run_or_local(
+        owner, team, project_name, run_uuid = get_project_run_or_local(
             project or ctx.obj.get("project"),
             uid,
             is_cli=True,
@@ -456,7 +460,9 @@ def get(ctx, project, uid, offline, path, output):
             )
             if output:
                 run_url = get_dashboard_url(
-                    subpath="{}/{}/runs/{}".format(owner, project_name, run_uuid)
+                    subpath="{}/runs/{}".format(
+                        get_project_subpath_url(owner, team, project_name), run_uuid
+                    )
                 )
                 config["url"] = run_url
                 handle_output(config, output)
@@ -529,7 +535,7 @@ def delete(ctx, project, uid, yes, offline, path):
         return
 
     # Resume normal flow
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -624,7 +630,7 @@ def update(ctx, project, uid, name, description, tags, offline, path):
             )
             sys.exit(1)
     else:
-        owner, project_name, run_uuid = get_project_run_or_local(
+        owner, _, project_name, run_uuid = get_project_run_or_local(
             project or ctx.obj.get("project"),
             uid or ctx.obj.get("run_uuid"),
             is_cli=True,
@@ -663,7 +669,7 @@ def approve(ctx, project, uid):
     \b
     $ polyaxon ops approve --uid 8aac02e3a62a4f0aaa257c59da5eab80
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -710,7 +716,7 @@ def stop(ctx, project, uid, yes):
     \b
     $ polyaxon ops stop --uid 8aac02e3a62a4f0aaa257c59da5eab80
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -762,7 +768,7 @@ def skip(ctx, project, uid, yes):
     \b
     $ polyaxon ops skip --uid 8aac02e3a62a4f0aaa257c59da5eab80
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -866,7 +872,7 @@ def restart(
     if polyaxonfile:
         content = OperationSpecification.read(polyaxonfile, is_preset=True).to_json()
 
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -951,7 +957,7 @@ def resume(
     if polyaxonfile:
         content = OperationSpecification.read(polyaxonfile, is_preset=True).to_json()
 
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -994,7 +1000,7 @@ def invalidate(ctx, project, uid):
     \b
     $ polyaxon ops invalidate --uid 8aac02e3a62a4f0aaa257c59da5eab80
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1183,7 +1189,7 @@ def execute(ctx, project, uid, executor):
                 message="Operation failed.\n{}".format(result["message"]),
             )
 
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1307,7 +1313,7 @@ def statuses(ctx, project, uid, watch, offline, path):
             sys.exit(1)
         return
 
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1375,7 +1381,7 @@ def statuses(ctx, project, uid, watch, offline, path):
 #             )
 #             sys.exit(1)
 #
-#     owner, project_name, run_uuid = get_project_run_or_local(
+#     owner, team, project_name, run_uuid = get_project_run_or_local(
 #         ctx.obj.get("project"), ctx.obj.get("run_uuid"), is_cli=True,
 #     )
 #
@@ -1467,7 +1473,7 @@ def logs(ctx, project, uid, follow, hide_time, all_containers, all_info, offline
             sys.exit(1)
         return
 
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1513,7 +1519,7 @@ def inspect(ctx, project, uid):
     \b
     $ polyaxon ops inspect -p acme/project -uid 8aac02e3a62a4f0aaa257c59da5eab80
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1572,7 +1578,7 @@ def shell(ctx, project, uid, command, pod, container):
     """
     from polyaxon._vendor.shell_pty import PseudoTerminal
 
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1677,7 +1683,7 @@ def artifacts(
     \b
     $ polyaxon ops artifacts -uid 8aac02e3a62a4f0aaa257c59da5eab80 -l-kind model -l-kind env --path="this/path"
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1855,7 +1861,7 @@ def upload(ctx, project, uid, path_from, path_to, sync_failure):
     \b
     $ polyaxon ops upload -uid 8aac02e3a62a4f0aaa257c59da5eab80 --path-to="path/to/upload/to"
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1929,7 +1935,7 @@ def transfer(ctx, project, uid, to_project):
     \b
     $ polyaxon ops transfer -p acme/foobar -uid 8aac02e3a62a4f0aaa257c59da5eab80 -to=dest-project
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, _, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -1997,13 +2003,16 @@ def dashboard(ctx, project, uid, yes, url, offline, path, server_config):
             sys.exit(1)
         run_data = RunConfigManager.read_from_path(offline_path)
         owner, project_name, run_uuid = run_data.owner, run_data.project, run_data.uuid
+        team = None
     else:
-        owner, project_name, run_uuid = get_project_run_or_local(
+        owner, team, project_name, run_uuid = get_project_run_or_local(
             project or ctx.obj.get("project"),
             uid or ctx.obj.get("run_uuid"),
             is_cli=True,
         )
-    subpath = "{}/{}/runs/{}".format(owner, project_name, run_uuid)
+    subpath = "{}/runs/{}".format(
+        get_project_subpath_url(owner, team, project_name), run_uuid
+    )
     dashboard_url = get_dashboard_url(subpath=subpath)
     get_dashboard(dashboard_url=dashboard_url, url_only=url, yes=yes)
     if offline:
@@ -2047,7 +2056,7 @@ def service(ctx, project, uid, yes, external, url):
     You can open the service embedded in Polyaxon UI or using the real service URL,
     please use the `--external` flag.
     """
-    owner, project_name, run_uuid = get_project_run_or_local(
+    owner, team, project_name, run_uuid = get_project_run_or_local(
         project or ctx.obj.get("project"),
         uid or ctx.obj.get("run_uuid"),
         is_cli=True,
@@ -2081,7 +2090,9 @@ def service(ctx, project, uid, yes, external, url):
     wait_for_running_condition(client)
 
     run_url = get_dashboard_url(
-        subpath="{}/{}/runs/{}/service".format(owner, project_name, run_uuid)
+        subpath="{}/runs/{}/service".format(
+            get_project_subpath_url(owner, team, project_name), run_uuid
+        )
     )
 
     namespace = client.run_data.settings.namespace
@@ -2091,8 +2102,8 @@ def service(ctx, project, uid, yes, external, url):
     if rewrite_path:
         service_endpoint = REWRITE_EXTERNAL_V1 if is_external else REWRITE_SERVICES_V1
 
-    service_subpath = "{}/{}/{}/runs/{}/".format(
-        namespace, owner, project_name, run_uuid
+    service_subpath = "{}/{}/runs/{}/".format(
+        get_project_subpath_url(namespace, team, owner), project_name, run_uuid
     )
     port = 80
     if client.settings and client.settings.agent:
@@ -2188,7 +2199,7 @@ def pull(
     \b
     $ polyaxon ops pull -a
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
 
@@ -2308,7 +2319,7 @@ def push(ctx, project, uid, all_runs, no_artifacts, clean, path, reset_project):
     \b
     $ polyaxon ops push -uid 8aac02e3a62a4f0aaa257c59da5eab80 --reset-project -p send-to-project
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
 

polyaxon/_cli/project_versions.py

@@ -12,7 +12,7 @@ from clipped.utils.responses import get_meta_response
 from clipped.utils.validation import validate_tags
 from urllib3.exceptions import HTTPError
 
-from polyaxon._cli.dashboard import get_dashboard_url
+from polyaxon._cli.dashboard import get_dashboard_url, get_project_subpath_url
 from polyaxon._cli.errors import handle_cli_error
 from polyaxon._contexts.paths import get_offline_base_path
 from polyaxon._schemas.lifecycle import V1ProjectVersionKind
@@ -170,6 +170,7 @@ def list_project_versions(
 
 def register_project_version(
     owner: str,
+    team: Optional[str],
     project_name: str,
     version: str,
     kind: V1ProjectVersionKind,
@@ -209,7 +210,9 @@ def register_project_version(
     Printer.print(
         "You can view this version on Polyaxon UI: {}".format(
             get_dashboard_url(
-                subpath="{}/{}/{}s/{}".format(owner, project_name, kind, version)
+                subpath="{}/{}s/{}".format(
+                    get_project_subpath_url(owner, team, project_name), kind, version
+                )
             )
         )
     )
@@ -217,6 +220,7 @@ def register_project_version(
 
 def copy_project_version(
     owner: str,
+    team: Optional[str],
     project_name: str,
     version: str,
     kind: V1ProjectVersionKind,
@@ -261,8 +265,10 @@ def copy_project_version(
     Printer.print(
         "You can view this version on Polyaxon UI: {}".format(
             get_dashboard_url(
-                subpath="{}/{}/{}s/{}".format(
-                    owner, to_project or project_name, kind, _version.name
+                subpath="{}/{}s/{}".format(
+                    get_project_subpath_url(owner, team, to_project or project_name),
+                    kind,
+                    _version.name,
                 )
             )
         )
@@ -271,6 +277,7 @@ def copy_project_version(
 
 def get_project_version(
     owner: str,
+    team: Optional[str],
     project_name: str,
     kind: V1ProjectVersionKind,
     version: str,
@@ -288,6 +295,17 @@ def get_project_version(
     try:
         response = polyaxon_client.get_version(kind, version)
         get_version_details(response, content_callback)
+        Printer.print(
+            "You can view this version on Polyaxon UI: {}".format(
+                get_dashboard_url(
+                    subpath="{}/{}s/{}".format(
+                        get_project_subpath_url(owner, team, project_name),
+                        kind,
+                        version,
+                    )
+                )
+            )
+        )
     except (ApiException, HTTPError) as e:
         handle_cli_error(
             e,
@@ -470,13 +488,16 @@ def stage_project_version(
 
 def open_project_version_dashboard(
     owner: str,
+    team: Optional[str],
     project_name: str,
     kind: V1ProjectVersionKind,
     version: str,
     url: str,
     yes: bool = False,
 ):
-    subpath = "{}/{}/{}s/{}".format(owner, project_name, kind, version)
+    subpath = "{}/{}s/{}".format(
+        get_project_subpath_url(owner, team, project_name), kind, version
+    )
 
     artifact_url = get_dashboard_url(subpath=subpath)
     if url:

polyaxon/_cli/projects.py

@@ -8,7 +8,7 @@ from clipped.utils.responses import get_meta_response
 from clipped.utils.validation import validate_tags
 from urllib3.exceptions import HTTPError
 
-from polyaxon._cli.dashboard import get_dashboard_url
+from polyaxon._cli.dashboard import get_dashboard_url, get_project_subpath_url
 from polyaxon._cli.errors import handle_cli_error
 from polyaxon._cli.init import init as init_project
 from polyaxon._cli.options import OPTIONS_NAME, OPTIONS_OWNER, OPTIONS_PROJECT
@@ -19,6 +19,7 @@ from polyaxon._env_vars.getters.user import get_local_owner
 from polyaxon._managers.project import ProjectConfigManager
 from polyaxon._utils import cache
 from polyaxon._utils.cache import get_local_project
+from polyaxon._utils.fqn_utils import get_owner_team_space
 from polyaxon.client import ProjectClient, V1Project
 from polyaxon.exceptions import ApiException
 from polyaxon.logger import clean_outputs
@@ -74,7 +75,7 @@ def create(ctx, name, description, tags, public, init):
             command_help="project create",
             sys_exit=True,
         )
-    owner, project_name = resolve_entity_info(
+    owner, team, project_name = resolve_entity_info(
         name or ctx.obj.get("project"), is_cli=True, entity_name="project"
     )
 
@@ -90,7 +91,9 @@ def create(ctx, name, description, tags, public, init):
         project_config = V1Project(
             name=project_name, description=description, tags=tags, is_public=public
         )
-        polyaxon_client = ProjectClient(owner=owner, manual_exceptions_handling=True)
+        polyaxon_client = ProjectClient(
+            owner=get_owner_team_space(owner, team), manual_exceptions_handling=True
+        )
         _project = polyaxon_client.create(project_config)
         config = polyaxon_client.client.sanitize_for_serialization(_project)
         cache.cache(config_manager=ProjectConfigManager, config=config)
@@ -103,7 +106,9 @@ def create(ctx, name, description, tags, public, init):
     Printer.success("Project `{}` was created successfully.".format(_project.name))
     Printer.print(
         "You can view this project on Polyaxon UI: {}".format(
-            get_dashboard_url(subpath="{}/{}".format(owner, _project.name))
+            get_dashboard_url(
+                subpath=get_project_subpath_url(owner, team, _project.name)
+            )
         )
     )
 
@@ -195,7 +200,7 @@ def get(ctx, _project):
     \b
     $ polyaxon project get -p owner/project
     """
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         _project or ctx.obj.get("project"), is_cli=True
     )
 
@@ -214,6 +219,13 @@ def get(ctx, _project):
             project=project_name,
         )
         get_entity_details(polyaxon_client.project_data, "Project")
+        Printer.print(
+            "You can view this project on Polyaxon UI: {}".format(
+                get_dashboard_url(
+                    subpath=get_project_subpath_url(owner, team, project_name)
+                )
+            )
+        )
     except (ApiException, HTTPError) as e:
         handle_cli_error(
             e, message="Could not get project `{}`.".format(project_name), sys_exit=True
@@ -237,7 +249,7 @@ def delete(ctx, _project, yes):
 
     Uses /docs/core/cli/#caching
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         _project or ctx.obj.get("project"), is_cli=True
     )
 
@@ -300,7 +312,7 @@ def update(ctx, _project, name, description, tags, private):
     \b
     $ polyaxon update --tags="foo, bar"
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         _project or ctx.obj.get("project"), is_cli=True
     )
 
@@ -357,10 +369,12 @@ def update(ctx, _project, name, description, tags, private):
 @clean_outputs
 def dashboard(ctx, _project, yes, url):
     """Open this project's dashboard details in browser."""
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         _project or ctx.obj.get("project"), is_cli=True
     )
-    project_url = get_dashboard_url(subpath="{}/{}".format(owner, project_name))
+    project_url = get_dashboard_url(
+        subpath=get_project_subpath_url(owner, team, project_name)
+    )
     if url:
         Printer.header("The dashboard is available at: {}".format(project_url))
         sys.exit(0)

polyaxon/_cli/run.py

@@ -10,7 +10,7 @@ from clipped.utils import git as git_utils
 from clipped.utils.validation import validate_tags
 from urllib3.exceptions import HTTPError
 
-from polyaxon._cli.dashboard import get_dashboard_url
+from polyaxon._cli.dashboard import get_dashboard_url, get_project_subpath_url
 from polyaxon._cli.errors import handle_cli_error
 from polyaxon._cli.operations import approve
 from polyaxon._cli.operations import execute as run_execute
@@ -30,6 +30,7 @@ from polyaxon._polyaxonfile import check_polyaxonfile
 from polyaxon._runner.kinds import RunnerKind
 from polyaxon._schemas.lifecycle import ManagedBy
 from polyaxon._utils import cache
+from polyaxon._utils.fqn_utils import get_owner_team_space
 from polyaxon.client import RunClient
 from polyaxon.exceptions import ApiException
 from polyaxon.logger import clean_outputs
@@ -43,6 +44,7 @@ def _run(
     ctx,
     name: str,
     owner: str,
+    team: Optional[str],
     project_name: str,
     description: str,
     tags: List[str],
@@ -61,11 +63,14 @@ def _run(
         owner=owner, project=project_name, manual_exceptions_handling=True
     )
 
+    owner_team = get_owner_team_space(owner, team)
+    project_url = get_project_subpath_url(owner, team, project_name)
+
     def get_instance_info(r):
         rn = (
             f"[white]{r.name}[/white]@[white]{r.uuid}[/white]"
             if r.name
-            else "[white]{r.uuid}[/white]"
+            else f"[white]{r.uuid}[/white]"
         )
         return f"[white]{owner}[/white]/[white]{project_name}[/white]:{rn}"
 
@@ -93,8 +98,15 @@ def _run(
                 meta_info=meta_info,
                 pending=pending,
             )
+            host_kwargs = {}
+            if (
+                response.settings
+                and response.settings.agent
+                and response.settings.agent.host
+            ):
+                host_kwargs["host"] = response.settings.agent.host
             run_url = get_dashboard_url(
-                subpath="{}/{}/runs/{}".format(owner, project_name, response.uuid)
+                subpath="{}/runs/{}".format(project_url, response.uuid), **host_kwargs
             )
             if output:
                 response_data = polyaxon_client.client.sanitize_for_serialization(
@@ -125,26 +137,38 @@ def _run(
 
     def execute_run_locally(run_uuid: str):
         ctx.obj = {
-            "project": "{}/{}".format(owner, project_name),
+            "project": "{}/{}".format(owner_team, project_name),
             "run_uuid": run_uuid,
             "executor": executor,
         }
         ctx.invoke(run_execute)
 
     def watch_run_statuses(run_uuid: str):
-        ctx.obj = {"project": "{}/{}".format(owner, project_name), "run_uuid": run_uuid}
+        ctx.obj = {
+            "project": "{}/{}".format(owner_team, project_name),
+            "run_uuid": run_uuid,
+        }
         ctx.invoke(statuses, watch=True)
 
     def watch_run_logs(run_uuid: str):
-        ctx.obj = {"project": "{}/{}".format(owner, project_name), "run_uuid": run_uuid}
+        ctx.obj = {
+            "project": "{}/{}".format(owner_team, project_name),
+            "run_uuid": run_uuid,
+        }
         ctx.invoke(run_logs)
 
     def start_run_shell(run_uuid: str):
-        ctx.obj = {"project": "{}/{}".format(owner, project_name), "run_uuid": run_uuid}
+        ctx.obj = {
+            "project": "{}/{}".format(owner_team, project_name),
+            "run_uuid": run_uuid,
+        }
         ctx.invoke(run_shell)
 
     def upload_run(run_uuid: str):
-        ctx.obj = {"project": "{}/{}".format(owner, project_name), "run_uuid": run_uuid}
+        ctx.obj = {
+            "project": "{}/{}".format(owner_team, project_name),
+            "run_uuid": run_uuid,
+        }
         ctx.invoke(
             run_upload, path_to=upload_to, path_from=upload_from, sync_failure=True
         )
@@ -555,13 +579,14 @@ def run(
         Printer.print("Please customize the specification or disable the template!")
         sys.exit(1)
 
-    owner, project_name = get_project_or_local(project, is_cli=True)
+    owner, team, project_name = get_project_or_local(project, is_cli=True)
     tags = validate_tags(tags, validate_yaml=True)
 
     _run(
         ctx=ctx,
         name=name,
         owner=owner,
+        team=team,
         project_name=project_name,
         description=description,
         tags=tags,

polyaxon/_client/mixin.py

@@ -0,0 +1,39 @@
+from polyaxon import settings
+from polyaxon._client.client import PolyaxonClient
+from polyaxon._schemas.client import ClientConfig
+from polyaxon._utils.fqn_utils import split_owner_team_space
+
+
+class ClientMixin:
+    @property
+    def client(self):
+        if self._client:
+            return self._client
+        self._client = PolyaxonClient()
+        return self._client
+
+    def reset_client(self, **kwargs):
+        if not settings.CLIENT_CONFIG.in_cluster:
+            self._client = PolyaxonClient(
+                ClientConfig.patch_from(settings.CLIENT_CONFIG, **kwargs)
+            )
+
+    @property
+    def owner(self) -> str:
+        return self._owner
+
+    @property
+    def team(self):
+        return self._team
+
+    @property
+    def project(self) -> str:
+        return self._project
+
+    def set_project(self, project: str):
+        self._project = project
+
+    def set_owner(self, owner: str):
+        owner, team = split_owner_team_space(owner)
+        self._owner = owner
+        self._team = team

polyaxon/_client/project.py

@@ -12,6 +12,7 @@ from clipped.utils.validation import validate_tags
 
 from polyaxon._client.client import PolyaxonClient
 from polyaxon._client.decorators import client_handler, get_global_or_inline_config
+from polyaxon._client.mixin import ClientMixin
 from polyaxon._constants.globals import DEFAULT
 from polyaxon._contexts import paths as ctx_paths
 from polyaxon._env_vars.getters.user import get_local_owner
@@ -21,13 +22,17 @@ from polyaxon._sdk.schemas.v1_list_project_versions_response import (
 )
 from polyaxon._sdk.schemas.v1_project import V1Project
 from polyaxon._sdk.schemas.v1_project_version import V1ProjectVersion
-from polyaxon._utils.fqn_utils import get_entity_full_name, get_entity_info
+from polyaxon._utils.fqn_utils import (
+    get_entity_full_name,
+    get_entity_info,
+    split_owner_team_space,
+)
 from polyaxon.exceptions import ApiException, PolyaxonClientException
 from polyaxon.logger import logger
 from traceml.artifacts import V1RunArtifact
 
 
-class ProjectClient:
+class ProjectClient(ClientMixin):
     """ProjectClient is a client to communicate with Polyaxon projects endpoints.
 
     If no values are passed to this class,
@@ -94,26 +99,13 @@ class ProjectClient:
         if not owner:
             raise PolyaxonClientException("Please provide a valid owner.")
 
+        owner, team = split_owner_team_space(owner)
         self._client = client
         self._owner = owner or DEFAULT
+        self._team = team
         self._project = project
         self._project_data = V1Project.construct()
 
-    @property
-    def client(self):
-        if self._client:
-            return self._client
-        self._client = PolyaxonClient()
-        return self._client
-
-    @property
-    def owner(self):
-        return self._owner
-
-    @property
-    def project(self):
-        return self._project
-
     @property
     def project_data(self):
         return self._project_data
@@ -139,11 +131,19 @@ class ProjectClient:
         Returns:
             V1Project, project instance from the response.
         """
-        self._project_data = self.client.projects_v1.create_project(
-            self.owner,
-            data,
-            async_req=False,
-        )
+        if self.team:
+            self._project_data = self.client.projects_v1.create_team_project(
+                self.owner,
+                self.team,
+                data,
+                async_req=False,
+            )
+        else:
+            self._project_data = self.client.projects_v1.create_project(
+                self.owner,
+                data,
+                async_req=False,
+            )
         self._project_data.owner = self.owner
         self._project = self._project_data.name
         return self._project_data