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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
polyaxon/_client/run.py CHANGED
@@ -35,7 +35,6 @@ from polyaxon import settings
35
35
  from polyaxon._cli.errors import handle_cli_error
36
36
  from polyaxon._client.client import PolyaxonClient
37
37
  from polyaxon._client.decorators import client_handler, get_global_or_inline_config
38
- from polyaxon._client.mixin import ClientMixin
39
38
  from polyaxon._client.store import PolyaxonStore
40
39
  from polyaxon._constants.metadata import META_COPY_ARTIFACTS, META_RECOMPILE
41
40
  from polyaxon._containers.names import MAIN_CONTAINER_NAMES
@@ -64,11 +63,7 @@ from polyaxon._sdk.schemas.v1_operation_body import V1OperationBody
64
63
  from polyaxon._sdk.schemas.v1_project_version import V1ProjectVersion
65
64
  from polyaxon._sdk.schemas.v1_run import V1Run
66
65
  from polyaxon._sdk.schemas.v1_run_settings import V1RunSettings
67
- from polyaxon._utils.fqn_utils import (
68
- get_entity_full_name,
69
- split_owner_team_space,
70
- to_fqn_name,
71
- )
66
+ from polyaxon._utils.fqn_utils import get_entity_full_name, to_fqn_name
72
67
  from polyaxon._utils.urls_utils import get_proxy_run_url
73
68
  from polyaxon.api import K8S_V1_LOCATION, STREAMS_V1_LOCATION
74
69
  from polyaxon.exceptions import ApiException, PolyaxonClientException
@@ -85,7 +80,7 @@ if TYPE_CHECKING:
85
80
  from traceml.tracking.run import Run
86
81
 
87
82
 
88
- class RunClient(ClientMixin):
83
+ class RunClient:
89
84
  """RunClient is a client to communicate with Polyaxon runs endpoints.
90
85
 
91
86
  If no values are passed to this class,
@@ -151,7 +146,7 @@ class RunClient(ClientMixin):
151
146
  return
152
147
 
153
148
  try:
154
- owner, _, project = get_project_or_local(
149
+ owner, project = get_project_or_local(
155
150
  get_entity_full_name(owner=owner, entity=project)
156
151
  )
157
152
  except PolyaxonClientException:
@@ -171,10 +166,8 @@ class RunClient(ClientMixin):
171
166
  if error_message and not self._is_offline:
172
167
  raise PolyaxonClientException(error_message)
173
168
 
174
- owner, team = split_owner_team_space(owner)
175
169
  self._client = client
176
170
  self._owner = owner
177
- self._team = team
178
171
  self._project = project
179
172
  self._run_uuid = (
180
173
  get_run_or_local(run_uuid)
@@ -225,11 +218,12 @@ class RunClient(ClientMixin):
225
218
  return client.config.no_op
226
219
  return settings.CLIENT_CONFIG.no_op
227
220
 
228
- def _use_agent_host(self):
229
- if self.settings.agent and self.settings.agent.url:
230
- self.reset_client(
231
- host=self.settings.agent.url, POLYAXON_HOST=self.settings.agent.url
232
- )
221
+ @property
222
+ def client(self):
223
+ if self._client:
224
+ return self._client
225
+ self._client = PolyaxonClient()
226
+ return self._client
233
227
 
234
228
  @property
235
229
  def store(self):
@@ -262,6 +256,20 @@ class RunClient(ClientMixin):
262
256
  return self.settings.artifacts_store.name
263
257
  return None
264
258
 
259
+ @property
260
+ def owner(self) -> str:
261
+ return self._owner or ""
262
+
263
+ def set_owner(self, owner: str):
264
+ self._owner = owner
265
+
266
+ @property
267
+ def project(self) -> str:
268
+ return self._project or ""
269
+
270
+ def set_project(self, project: str):
271
+ self._project = project
272
+
265
273
  @property
266
274
  def run_uuid(self) -> str:
267
275
  return self._run_uuid
@@ -890,7 +898,6 @@ class RunClient(ClientMixin):
890
898
  """
891
899
  if not self.settings:
892
900
  self.refresh_data()
893
- self._use_agent_host()
894
901
  params = get_logs_params(
895
902
  last_file=last_file, last_time=last_time, connection=self.artifacts_store
896
903
  )
@@ -914,7 +921,6 @@ class RunClient(ClientMixin):
914
921
  def inspect(self):
915
922
  if not self.settings:
916
923
  self.refresh_data()
917
- self._use_agent_host()
918
924
  params = get_streams_params(connection=self.artifacts_store, status=self.status)
919
925
  return self.client.runs_v1.inspect_run(
920
926
  self.namespace, self.owner, self.project, self.run_uuid, **params
@@ -988,10 +994,6 @@ class RunClient(ClientMixin):
988
994
  if not container:
989
995
  container = pod_containers[0]
990
996
 
991
- if not self.settings:
992
- self.refresh_data()
993
- self._use_agent_host()
994
-
995
997
  url = get_proxy_run_url(
996
998
  service=K8S_V1_LOCATION,
997
999
  namespace=self.namespace,
@@ -1037,8 +1039,6 @@ class RunClient(ClientMixin):
1037
1039
  """
1038
1040
  if not self.settings:
1039
1041
  self.refresh_data()
1040
- self._use_agent_host()
1041
-
1042
1042
  params = get_streams_params(self.artifacts_store)
1043
1043
  return self.client.runs_v1.get_run_events(
1044
1044
  self.namespace,
@@ -1074,8 +1074,6 @@ class RunClient(ClientMixin):
1074
1074
  """
1075
1075
  if not self.settings:
1076
1076
  self.refresh_data()
1077
- self._use_agent_host()
1078
-
1079
1077
  params = get_streams_params(self.artifacts_store)
1080
1078
  return self.client.runs_v1.get_multi_run_events(
1081
1079
  self.namespace,
@@ -1312,7 +1310,6 @@ class RunClient(ClientMixin):
1312
1310
  """
1313
1311
  if not self.settings:
1314
1312
  self.refresh_data()
1315
- self._use_agent_host()
1316
1313
  params = get_streams_params(self.artifacts_store)
1317
1314
  return self.client.runs_v1.get_run_artifact(
1318
1315
  namespace=self.namespace,
@@ -1346,10 +1343,6 @@ class RunClient(ClientMixin):
1346
1343
  if not self.run_uuid:
1347
1344
  return
1348
1345
 
1349
- if not self.settings:
1350
- self.refresh_data()
1351
- self._use_agent_host()
1352
-
1353
1346
  lineage_path = lineage.path or ""
1354
1347
  summary = lineage.summary or {}
1355
1348
  is_event = summary.get("is_event")
@@ -1428,8 +1421,6 @@ class RunClient(ClientMixin):
1428
1421
  """
1429
1422
  if not self.settings:
1430
1423
  self.refresh_data()
1431
- self._use_agent_host()
1432
-
1433
1424
  url = get_proxy_run_url(
1434
1425
  service=STREAMS_V1_LOCATION,
1435
1426
  namespace=self.namespace,
@@ -1469,8 +1460,6 @@ class RunClient(ClientMixin):
1469
1460
  """
1470
1461
  if not self.settings:
1471
1462
  self.refresh_data()
1472
- self._use_agent_host()
1473
-
1474
1463
  url = get_proxy_run_url(
1475
1464
  service=STREAMS_V1_LOCATION,
1476
1465
  namespace=self.namespace,
@@ -1518,7 +1507,6 @@ class RunClient(ClientMixin):
1518
1507
  """
1519
1508
  if not self.settings:
1520
1509
  self.refresh_data()
1521
- self._use_agent_host()
1522
1510
 
1523
1511
  params = get_streams_params(connection=self.artifacts_store)
1524
1512
  url = get_proxy_run_url(
@@ -1608,7 +1596,6 @@ class RunClient(ClientMixin):
1608
1596
 
1609
1597
  if not self.settings:
1610
1598
  self.refresh_data()
1611
- self._use_agent_host()
1612
1599
 
1613
1600
  params = get_streams_params(connection=self.artifacts_store)
1614
1601
  url = get_proxy_run_url(
@@ -1639,8 +1626,6 @@ class RunClient(ClientMixin):
1639
1626
  """
1640
1627
  if not self.settings:
1641
1628
  self.refresh_data()
1642
- self._use_agent_host()
1643
-
1644
1629
  params = get_streams_params(connection=self.artifacts_store)
1645
1630
  self.client.runs_v1.delete_run_artifact(
1646
1631
  namespace=self.namespace,
@@ -1660,8 +1645,6 @@ class RunClient(ClientMixin):
1660
1645
  """
1661
1646
  if not self.settings:
1662
1647
  self.refresh_data()
1663
- self._use_agent_host()
1664
-
1665
1648
  params = get_streams_params(connection=self.artifacts_store)
1666
1649
  return self.client.runs_v1.delete_run_artifacts(
1667
1650
  namespace=self.namespace,
@@ -1684,8 +1667,6 @@ class RunClient(ClientMixin):
1684
1667
  """
1685
1668
  if not self.settings:
1686
1669
  self.refresh_data()
1687
- self._use_agent_host()
1688
-
1689
1670
  params = get_streams_params(connection=self.artifacts_store)
1690
1671
  return self.client.runs_v1.get_run_artifacts_tree(
1691
1672
  namespace=self.namespace,
@@ -33,9 +33,7 @@ class RayJobContextsManager(BaseContextsManager):
33
33
  connection_by_names=connection_by_names,
34
34
  )
35
35
 
36
- data = {
36
+ return {
37
37
  "head": _get_replica(job.head),
38
+ "workers": {wn: _get_replica(job.workers[wn]) for wn in job.workers},
38
39
  }
39
- if job.workers:
40
- data["workers"] = {wn: _get_replica(job.workers[wn]) for wn in job.workers}
41
- return data
@@ -5,7 +5,7 @@ from clipped.utils.strings import validate_slug
5
5
 
6
6
  from polyaxon._constants.globals import DEFAULT
7
7
  from polyaxon._env_vars.getters.user import get_local_owner
8
- from polyaxon._utils.fqn_utils import get_entity_info, split_owner_team_space
8
+ from polyaxon._utils.fqn_utils import get_entity_info
9
9
  from polyaxon.exceptions import PolyaxonClientException, PolyaxonSchemaError
10
10
 
11
11
 
@@ -31,8 +31,6 @@ def resolve_entity_info(entity: str, entity_name: str, is_cli: bool = False):
31
31
  if not owner:
32
32
  owner = settings.AUTH_CONFIG.username if settings.AUTH_CONFIG else None
33
33
 
34
- owner, team = split_owner_team_space(owner)
35
-
36
34
  if not all([owner, entity_value]):
37
35
  message = "Please provide a valid {}.".format(entity_name)
38
36
  if is_cli:
@@ -51,4 +49,4 @@ def resolve_entity_info(entity: str, entity_name: str, is_cli: bool = False):
51
49
  entity_name, entity_value
52
50
  )
53
51
  )
54
- return owner, team, entity_value
52
+ return owner, entity_value
@@ -7,7 +7,7 @@ from polyaxon._constants.globals import DEFAULT
7
7
  from polyaxon._env_vars.getters.user import get_local_owner
8
8
  from polyaxon._managers.project import ProjectConfigManager
9
9
  from polyaxon._utils.cache import get_local_project
10
- from polyaxon._utils.fqn_utils import get_entity_info, split_owner_team_space
10
+ from polyaxon._utils.fqn_utils import get_entity_info
11
11
  from polyaxon.exceptions import PolyaxonClientException, PolyaxonSchemaError
12
12
 
13
13
 
@@ -52,8 +52,6 @@ def get_project_or_local(project=None, is_cli: bool = False):
52
52
  if not owner and (not settings.CLI_CONFIG or settings.CLI_CONFIG.is_community):
53
53
  owner = DEFAULT
54
54
 
55
- owner, team = split_owner_team_space(owner)
56
-
57
55
  if not all([owner, project_name]):
58
56
  error_message = get_project_error_message(owner, project_name)
59
57
  if is_cli:
@@ -81,4 +79,4 @@ def get_project_or_local(project=None, is_cli: bool = False):
81
79
  sys.exit(1)
82
80
  else:
83
81
  raise PolyaxonSchemaError(error_message)
84
- return owner, team, project_name
82
+ return owner, project_name
@@ -37,9 +37,9 @@ def get_run_or_local(run_uuid=None, is_cli: bool = False):
37
37
 
38
38
 
39
39
  def get_project_run_or_local(project=None, run_uuid=None, is_cli: bool = True):
40
- owner, team, project_name = get_project_or_local(project, is_cli=is_cli)
40
+ user, project_name = get_project_or_local(project, is_cli=is_cli)
41
41
  run_uuid = get_run_or_local(run_uuid, is_cli=is_cli)
42
- return owner, team, project_name, run_uuid
42
+ return user, project_name, run_uuid
43
43
 
44
44
 
45
45
  def get_collect_artifacts(arg: Optional[bool] = None, default: Optional[bool] = None):
@@ -37,9 +37,6 @@ def requests_gpu(resources: k8s_schemas.V1ResourceRequirements) -> bool:
37
37
  if not resources:
38
38
  return False
39
39
 
40
- if not isinstance(resources, k8s_schemas.V1ResourceRequirements):
41
- resources = k8s_schemas.V1ResourceRequirements(**resources)
42
-
43
40
  if resources.requests:
44
41
  for key, val in resources.requests.items():
45
42
  if "gpu" in key and val is not None and val > 0:
@@ -58,9 +58,7 @@ class RayJobConverter(RayJobMixin, BaseConverter):
58
58
  config=compiled_operation.plugins, auth=default_auth
59
59
  )
60
60
  head = _get_replica(job.head)
61
- workers = None
62
- if job.workers:
63
- workers = {n: _get_replica(w) for n, w in job.workers.items()}
61
+ workers = {n: _get_replica(w) for n, w in job.workers.items()}
64
62
  labels = self.get_labels(version=pkg.VERSION, labels={})
65
63
 
66
64
  return get_ray_job_custom_resource(
@@ -70,7 +68,7 @@ class RayJobConverter(RayJobMixin, BaseConverter):
70
68
  workers=workers,
71
69
  entrypoint=job.entrypoint,
72
70
  metadata=job.metadata,
73
- runtime_env=orjson_dumps(job.runtime_env),
71
+ runtime_env=encode(orjson_dumps(job.runtime_env)),
74
72
  ray_version=job.ray_version,
75
73
  termination=compiled_operation.termination,
76
74
  collect_logs=plugins.collect_logs,
@@ -46,7 +46,7 @@ def set_notify(custom_object: Dict, notifications: List[V1Notification]) -> Dict
46
46
  def set_clean_pod_policy(template_spec: Dict, clean_pod_policy: str) -> Dict:
47
47
  if not clean_pod_policy:
48
48
  # Sets default clean pod policy
49
- clean_pod_policy = "None"
49
+ clean_pod_policy = "All"
50
50
 
51
51
  template_spec["cleanPodPolicy"] = clean_pod_policy.capitalize()
52
52
  return template_spec
@@ -9,4 +9,3 @@ class V1SettingsCatalog(BaseAllowSchemaModel):
9
9
  uuid: Optional[UUIDStr]
10
10
  name: Optional[StrictStr]
11
11
  version: Optional[StrictStr]
12
- url: Optional[StrictStr]
@@ -17,7 +17,6 @@ class V1Team(BaseAllowSchemaModel):
17
17
  component_hubs: Optional[List[StrictStr]]
18
18
  model_registries: Optional[List[StrictStr]]
19
19
  settings: Optional[V1TeamSettings]
20
- policy: Optional[StrictStr]
21
20
  role: Optional[StrictStr]
22
21
  created_at: Optional[datetime.datetime]
23
22
  updated_at: Optional[datetime.datetime]
@@ -9,32 +9,15 @@ from clipped.utils.paths import get_relative_path_to
9
9
  from polyaxon.exceptions import PolyaxonSchemaError
10
10
 
11
11
 
12
- def get_owner_team_space(owner: str, team: Optional[str] = None) -> str:
13
- if team:
14
- return "{}/{}".format(owner, team)
15
- return owner
16
-
17
-
18
- def split_owner_team_space(owner: str) -> Tuple[str, Optional[str]]:
19
- return owner.split("/") if owner and "/" in owner else (owner, None)
20
-
21
-
22
- def _remove_team(owner: str) -> str:
23
- return split_owner_team_space(owner)[0]
24
-
25
-
26
12
  def get_project_instance(owner: str, project: str) -> str:
27
- owner = _remove_team(owner)
28
13
  return "{}.{}".format(owner, project)
29
14
 
30
15
 
31
16
  def get_run_instance(owner: str, project: str, run_uuid: str) -> str:
32
- owner = _remove_team(owner)
33
17
  return "{}.{}.runs.{}".format(owner, project, run_uuid)
34
18
 
35
19
 
36
20
  def get_cleaner_instance(owner: str, project: str, run_uuid: str) -> str:
37
- owner = _remove_team(owner)
38
21
  return "{}.{}.cleaners.{}".format(owner, project, run_uuid)
39
22
 
40
23
 
@@ -70,7 +53,6 @@ def to_fqn_name(name: str) -> str:
70
53
  def get_entity_full_name(
71
54
  owner: Optional[str] = None, entity: Optional[str] = None
72
55
  ) -> str:
73
- owner = _remove_team(owner)
74
56
  if owner and entity:
75
57
  return "{}/{}".format(owner, entity)
76
58
  return entity
@@ -83,14 +65,11 @@ def get_entity_info(entity: str) -> Tuple[str, str]:
83
65
  )
84
66
 
85
67
  parts = entity.replace(".", "/").split("/")
86
- if len(parts) > 3:
68
+ if len(parts) > 2:
87
69
  raise PolyaxonSchemaError(
88
70
  "Received an invalid entity reference: `{}`".format(entity)
89
71
  )
90
- if len(parts) == 3:
91
- owner = "/".join(parts[:2])
92
- entity_name = parts[2]
93
- elif len(parts) == 2:
72
+ if len(parts) == 2:
94
73
  owner, entity_name = parts
95
74
  else:
96
75
  owner = None
@@ -102,8 +81,6 @@ def get_entity_info(entity: str) -> Tuple[str, str]:
102
81
  def get_versioned_entity_full_name(
103
82
  owner: Optional[str], component: str, tag: Optional[str] = None
104
83
  ) -> str:
105
- owner = _remove_team(owner)
106
-
107
84
  if tag:
108
85
  component = "{}:{}".format(component, tag)
109
86
  if owner:
polyaxon/pkg.py CHANGED
@@ -1,5 +1,5 @@
1
1
  NAME = "polyaxon"
2
- VERSION = "2.2.0"
2
+ VERSION = "2.2.0-rc0"
3
3
  SCHEMA_VERSION = 1.1
4
4
  DESC = "Command Line Interface (CLI) and client to interact with Polyaxon API."
5
5
  URL = "https://github.com/polyaxon/polyaxon"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: polyaxon
3
- Version: 2.2.0
3
+ Version: 2.2.0rc0
4
4
  Summary: Command Line Interface (CLI) and client to interact with Polyaxon API.
5
5
  Home-page: https://github.com/polyaxon/polyaxon
6
6
  Author: Polyaxon, Inc.
@@ -9,7 +9,7 @@ polyaxon/exceptions.py,sha256=ujvG9p1Pn2KHYbHqB3-faadW46dEuULUQXNtfkd2zk8,10236
9
9
  polyaxon/fs.py,sha256=RS8XmVrrfXfIJXN6cTCCRRYwesCLHVVfC01Vi56lecs,246
10
10
  polyaxon/k8s.py,sha256=nI5oPCSlqU4aaeVShM6SlYS9eqYiYUL4GDXIZ4bnq-I,1051
11
11
  polyaxon/logger.py,sha256=gdZQms37Pe5G2j-Ear5jbSAJeGgX6fnvg7oE8_9MSlc,2309
12
- polyaxon/pkg.py,sha256=hbHWZ5qW4v8e3v9_0M8SgTf_2n-aISqDfaGd3GOQ8q4,262
12
+ polyaxon/pkg.py,sha256=yPU9X8AnlId98xXQ00GjqnAbU5UDzN1yCTcwRyOSYtU,266
13
13
  polyaxon/polyaxonfile.py,sha256=xHmHT_cHomfuAQm82Jhnp71YNN5mQ-Lod7EbonjY4b4,429
14
14
  polyaxon/schemas.py,sha256=-CykY3emoAUCs_zRNbjwkuMkqbaEDjfKsZC86rI8154,5870
15
15
  polyaxon/settings.py,sha256=Pxx1-T2oeJ5XmvGFN0YgnVzum_9FyTPaQtl68aQvYc4,4116
@@ -23,22 +23,22 @@ polyaxon/_auxiliaries/sidecar.py,sha256=XqAPteBHlSZbXPn_HKBEA5ar2IfcNANqfhe8lHrt
23
23
  polyaxon/_auxiliaries/tuner.py,sha256=W27fyK5PDwTSv3z-0nmvmw_dUM3UIoV5JCCNeK9CIKw,941
24
24
  polyaxon/_cli/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
25
25
  polyaxon/_cli/admin.py,sha256=xvnQyLKwqyxIzBV7_Xv59Z95-AAxU_h3qdu9UMg8AkE,8872
26
- polyaxon/_cli/artifacts.py,sha256=MLNqIcSNeKnCJVGz_XjsDz28zEZhp1i9EE7iNbLl1zs,18474
26
+ polyaxon/_cli/artifacts.py,sha256=wyexlRictltdRlhGj0nu-UjUd9XD2UQcBSN0TP934cA,18350
27
27
  polyaxon/_cli/auth.py,sha256=Ytvw3ik6_LiSICfZu6Qw5IM95Df7eelOeRFD-tu8EUE,5002
28
28
  polyaxon/_cli/check.py,sha256=WSQpQp3enPmgIXrllwgUTd3VcjYCG8P-37ONt7F67wo,1337
29
29
  polyaxon/_cli/completion.py,sha256=soJlrKTAFdxskWNH3YjgCMTM__NqVCr9Rr6z8FSmXao,2017
30
- polyaxon/_cli/components.py,sha256=pvfZB9NIjwA8qtfdebKWMAbdx--7ARV0FZImPnW6u4A,19818
31
- polyaxon/_cli/config.py,sha256=-ruIopLsC0ArPvw8RHlY5uQDu2_eaQ5PnvEDNJsq9f8,9165
32
- polyaxon/_cli/dashboard.py,sha256=zRrOJMfowM7YQnYXhz_guKsTi-jkoCuVl_g0UTJ37qE,1681
30
+ polyaxon/_cli/components.py,sha256=h4lot3iD_uLeT85-iR1dVaDNshleKvwzaSJ5SFdVDXM,19694
31
+ polyaxon/_cli/config.py,sha256=popMLIN_wYCMEY9S5N5NepiyTn9-oYf0TALtsb8cTX8,8278
32
+ polyaxon/_cli/dashboard.py,sha256=PrBVmxrT4GdsqXLWNDVysmYUIjUcixfYC7EM2LurQAI,1374
33
33
  polyaxon/_cli/errors.py,sha256=BYs7-I0CLLNYma-I0eJXB1EHs8ZTs_nWGRKGJz6CzTI,1031
34
- polyaxon/_cli/init.py,sha256=lZhkIbaHGv4FdMypdmXhYXRrH5GpM-qcDEks4IsmKIM,7714
35
- polyaxon/_cli/models.py,sha256=71oxGrIe4nQORnnfAiNi6T7A0DBnEe3V6whBABBGL3c,18066
36
- polyaxon/_cli/operations.py,sha256=iQTVHkVvUxFCRlaKBLP0xVxuxcb9KJHgTrtNe4Rf8LM,75624
34
+ polyaxon/_cli/init.py,sha256=vpi-fI_kK4bWldC1WtmrUp-N6AQ_LPSpaQkRfpqyGCk,7711
35
+ polyaxon/_cli/models.py,sha256=yyd5T-oJ_rxTAEiZNjiX-uCYh3R3JdxR1EJYQYdxeSU,17942
36
+ polyaxon/_cli/operations.py,sha256=g1zmhDqaKNpzVlxfKXrRzFbZTR2uxXPMJ4RUlfeQ2YM,74918
37
37
  polyaxon/_cli/options.py,sha256=-jeMZsdfg0JOV_QzVDf1hAhqK55NI0dkC_x4MZZWty8,1927
38
38
  polyaxon/_cli/port_forward.py,sha256=Lshpcrv7-4tXcriHmppiFW_3QZ7ZosDtUbJDIvdddSA,2733
39
- polyaxon/_cli/project_versions.py,sha256=fbgE3tRShrgH8TAA6ETj78J38HG2-BcVoAWbxYS9d5s,20507
40
- polyaxon/_cli/projects.py,sha256=RVuTbKtNP_bvw3kRFKaIQBWq0xUwF0BUHK19T87VG5Q,12069
41
- polyaxon/_cli/run.py,sha256=3KM_E0iLw8WUe36zeXuwMA5urzGljo6yf4A-DL1uz6k,17737
39
+ polyaxon/_cli/project_versions.py,sha256=aAwWuNsFTqH7mzofGMcomLqKcVGSfvbfX6VGryP2poE,19829
40
+ polyaxon/_cli/projects.py,sha256=CtjGSXUjUhvhKOdOJAWxS5r-5PpHeqgXbB6_F7wEZb8,11591
41
+ polyaxon/_cli/run.py,sha256=11j7_EGhMOB0dRGjczW_ldr251WSyPZqD7Qi61AnHys,17194
42
42
  polyaxon/_cli/session.py,sha256=5Plolpxwkv3wiup0pRYBpBPD64dvb4GzeRPz_nuIGRo,6033
43
43
  polyaxon/_cli/utils.py,sha256=jRD1npuD89-0SxtzAx5SE3i4Lvhi8vnfYkmyOMbDxEY,1319
44
44
  polyaxon/_cli/version.py,sha256=bC0T8S6Momr7EvHGV9Swn868x9ZHD7oL8ahQv57yInk,4536
@@ -55,9 +55,8 @@ polyaxon/_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
55
55
  polyaxon/_client/client.py,sha256=wLRAkm05qNi-rc5RpkWic3oJZNF0HTxkrtOOkbB0YqE,8015
56
56
  polyaxon/_client/impersonate.py,sha256=4jRoJoX8nkwvVc3zAYqVqPoysNT1kZoPRbwyTpyHGHw,1260
57
57
  polyaxon/_client/init.py,sha256=QECGjzuBTFaagndubzd9U1COpc5NGe-E0aVkSRetcIs,418
58
- polyaxon/_client/mixin.py,sha256=-tPm8RBej0UeG9HAKH3qN2fPX8PBtuXAHY1OpwLIDu4,1017
59
- polyaxon/_client/project.py,sha256=-vHLae0fMYqOjMscRm0I25Ixj-hS6o9i6XQm5PIxyiA,65520
60
- polyaxon/_client/run.py,sha256=7aUowwMB_yDCFYrlBdYuh-LCKcsg7IUbSdaLusoYBX8,118798
58
+ polyaxon/_client/project.py,sha256=Vw-Ve09Hw5h93-OJzSv6K7Z8eiCy6RHA-Jb6Ft7flkE,65375
59
+ polyaxon/_client/run.py,sha256=GV5s4e4HOw7p46QhOngkbiNfAtjzMW-EFShQs_QzU-E,118286
61
60
  polyaxon/_client/store.py,sha256=-Y33ypkijGQnhHTQ_vCTqLlpk0wRqoaP-ntJhdUtv7E,11311
62
61
  polyaxon/_client/decorators/__init__.py,sha256=e5CBijciLP-Ic-YkaL4tFhUdr--uod_TexvxAJamGZQ,186
63
62
  polyaxon/_client/decorators/client_call_handler.py,sha256=dgdidWsL9e8AgJmOfxz4urkI3bBgX64whWI1WHeAJDU,4587
@@ -80,7 +79,7 @@ polyaxon/_compiler/contexts/base.py,sha256=4EVQQlLgxvFORZ27ouBhwcJKhl3vMGXyOVP9H
80
79
  polyaxon/_compiler/contexts/contexts.py,sha256=tJy8eL9UrUhjisJ3MdYNmMVERSK_m_fKfgPTQgC-0y4,6535
81
80
  polyaxon/_compiler/contexts/dask_job.py,sha256=GxhLcZtCDWSDfc45UenmKsJ6Fhr0f_GeneaAzLHtncg,1265
82
81
  polyaxon/_compiler/contexts/job.py,sha256=1lkjfMO2eHndmmtxC551K6n6RE5XasqkWBfwNpYy_MA,759
83
- polyaxon/_compiler/contexts/ray_job.py,sha256=6CFFuX63HhyU0z--GoE3Tv2vENBB1QKGCwuFmEwC2PY,1292
82
+ polyaxon/_compiler/contexts/ray_job.py,sha256=rdf__86SFoUEYf5T3vUTZXGLL-QI10enYwyngXhEJ7Q,1242
84
83
  polyaxon/_compiler/contexts/service.py,sha256=d4zKB608906CXRZeUeOLgOdro5CEFe9byHWrGyy1ZFw,2150
85
84
  polyaxon/_compiler/contexts/kubeflow/__init__.py,sha256=FnOOO9l-eunvun5kGcTfQ9uO8gLhuHIFuh2_gCHxaLA,488
86
85
  polyaxon/_compiler/contexts/kubeflow/mpi_job.py,sha256=UMf7yJPRWEB9MVYS4Eoa7V6IZEQasU5VU71WdNZQ-2k,1214
@@ -169,10 +168,10 @@ polyaxon/_env_vars/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVG
169
168
  polyaxon/_env_vars/keys.py,sha256=DxRrXFQpiBCnve7tY3I9eeOf7tBvlrt92-XdH50__rk,7586
170
169
  polyaxon/_env_vars/getters/__init__.py,sha256=K59-CiBIqf8dDzvYBV3gL4AREidI9fe8JljFHyWxhtY,711
171
170
  polyaxon/_env_vars/getters/agent.py,sha256=dh-kIp1upD9DE_nTpBktQTxxEy7rZuh50u9UmRCWurg,1061
172
- polyaxon/_env_vars/getters/owner_entity.py,sha256=IIFVw3tYZtQhieZ8WjSlIx7Z5Q8XQogStzrw703XQn0,1766
173
- polyaxon/_env_vars/getters/project.py,sha256=5AoC2jxBe1x6rncGahevAe9J1cffYlCRdcpsRMa-2Nc,2762
171
+ polyaxon/_env_vars/getters/owner_entity.py,sha256=AhKDeu6ndnnzFmKWPMbjiBWFSd9H_a3dYvnGSEFHeZY,1687
172
+ polyaxon/_env_vars/getters/project.py,sha256=wTWPhfqonf9nv81nLeyZM18cMPudW4ljnfy2AZP55-Q,2683
174
173
  polyaxon/_env_vars/getters/queue.py,sha256=Ik2u1yVIwZZPM2InKvg9EFd4CmOxv2ZOMJt7UD1ve0g,572
175
- polyaxon/_env_vars/getters/run.py,sha256=HhE5Cl9-MvPLtNuVoOZEEaN6mMk9HPedQeEbb2zJzFk,2721
174
+ polyaxon/_env_vars/getters/run.py,sha256=y1GYhLIVyHiU2A4EqlpM3gu7MX2Ml-yd7holxHKK_k4,2707
176
175
  polyaxon/_env_vars/getters/user.py,sha256=Ey1h9o6I8uMDYzP0MKnpURUaRwQ0FHd4qvkgdtwO4GE,1134
177
176
  polyaxon/_env_vars/getters/versioned_entity.py,sha256=cQ8WYZ_yTBTw-v_26EKnSkLv1Au9Z1ZSH6gB-Y68pCc,2885
178
177
  polyaxon/_flow/__init__.py,sha256=j5h-jGgno0NczmhiPwri9BdkRZxpG3OC3AJSy_yw7gg,2842
@@ -303,14 +302,14 @@ polyaxon/_k8s/converter/base/main.py,sha256=eT5MlalYQ8U1QlvoBQp2EME6Na6sI4o_9ZTE
303
302
  polyaxon/_k8s/converter/base/mounts.py,sha256=YKJ-ifAZHg4lfo_ZWlBHoTBmyOt3wbT5Kx8SLBd21s8,3563
304
303
  polyaxon/_k8s/converter/base/sidecar.py,sha256=hrUWVKr1ijvmAOd0yZHxNJRpODeOsOvkCqEPS4XowjA,5965
305
304
  polyaxon/_k8s/converter/common/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
306
- polyaxon/_k8s/converter/common/accelerators.py,sha256=SGZrbfS3VN7aGdjZ6zwnRPUdYRUDsoIo1hiLNiz8QbQ,1425
305
+ polyaxon/_k8s/converter/common/accelerators.py,sha256=qMem6o4wz3EWZmd-usEr7LhyFx6GSSobj_7p8DXffNI,1286
307
306
  polyaxon/_k8s/converter/common/annotations.py,sha256=QV7LSxt2ezcu4T4uwNgdrACG1ptcD0H89FQIhSYfUEI,1228
308
307
  polyaxon/_k8s/converter/common/volumes.py,sha256=vjvTCF247hu1uETmRcrtK_sCSGABwaOJ3LkO5s7Os0U,3541
309
308
  polyaxon/_k8s/converter/converters/__init__.py,sha256=jyw6PnVQcG-lfyNbfxomFhh1aa2awv8O-D7YIaNduY8,1177
310
309
  polyaxon/_k8s/converter/converters/dask_job.py,sha256=DNlQczpN29AMJzsAy3q6SlpIH9h-kfcp3UMvfRLGJm8,2784
311
310
  polyaxon/_k8s/converter/converters/helpers.py,sha256=LEfPISHptnC_1W-B5lY3D834U0qAr6wsX4x1EafWlk4,661
312
311
  polyaxon/_k8s/converter/converters/job.py,sha256=luYWU9-qkuT5usk42zmd2JkX8thwWHNcqrmngtbHPJ8,2287
313
- polyaxon/_k8s/converter/converters/ray_job.py,sha256=d57kxNxt3VqN5Bu54NKFqH-HHyv0F2Ps74qJTNVNxSk,3264
312
+ polyaxon/_k8s/converter/converters/ray_job.py,sha256=gmoHJ0lyeXU1QMpCsptU2ur_T16zxuLAhh3G8KCVYQ0,3221
314
313
  polyaxon/_k8s/converter/converters/service.py,sha256=F2tW2bTlpaX9J47I8nE6up8xwUbdg2dQ_oXqsJdcdu8,2465
315
314
  polyaxon/_k8s/converter/converters/kubeflow/__init__.py,sha256=ZXdpmleFm0ALmJYPlU9JG6ZpsBW6bq6DvF1d9IVoW68,498
316
315
  polyaxon/_k8s/converter/converters/kubeflow/mpi_job.py,sha256=UyLqLBCmjGIWlj6ZiDJqMByzeVWTKp1wBtu4oogfL3w,2731
@@ -329,7 +328,7 @@ polyaxon/_k8s/custom_resources/job.py,sha256=hTETu_oU90y9CujwflodopZ_HMqi3ajMPsy
329
328
  polyaxon/_k8s/custom_resources/operation.py,sha256=qrtxZ6i19N5tZIYKqGRoqCM3dvQg5lSVC1mZMo0Kx3w,625
330
329
  polyaxon/_k8s/custom_resources/ray_job.py,sha256=Ytvsiaev4xSL5Y3SN8vcI15DZbStkOZJq11GaZa-T1g,6210
331
330
  polyaxon/_k8s/custom_resources/service.py,sha256=M_mptT7JR6jmuctfgo87ov7KXRM4_jggJ5nc3zXL7hA,2370
332
- polyaxon/_k8s/custom_resources/setter.py,sha256=WCbEf7DaijK2U-FNAPAVI7mEefA7rb-KhHYnWPRX6aU,1846
331
+ polyaxon/_k8s/custom_resources/setter.py,sha256=YgTV0vnTEHMibISHKkt3IRA6cozW-6ynXw8fGfcALpo,1845
333
332
  polyaxon/_k8s/custom_resources/kubeflow/__init__.py,sha256=oCDo38kAVl7L1ehQTfWX04fZHRTa-V5qYKbef9-jG7k,556
334
333
  polyaxon/_k8s/custom_resources/kubeflow/common.py,sha256=_OrM906v50aTziJcGUOQqzr5XYBuTcNeErgAau2ukng,1041
335
334
  polyaxon/_k8s/custom_resources/kubeflow/mpi_job.py,sha256=GwBfAb4LVYvDdHdzBUW2QgCq-_ypEhKyPrkInkaXyZo,2565
@@ -541,9 +540,9 @@ polyaxon/_sdk/schemas/v1_search.py,sha256=zz1R5x4FEWBSEW18uAbgV4Uea7O4KU7VM_Usm2
541
540
  polyaxon/_sdk/schemas/v1_search_spec.py,sha256=mFGNgsbXgSiLx08i9-vM0jsyR3BBGqAxBoE68RV-Rbg,788
542
541
  polyaxon/_sdk/schemas/v1_section_spec.py,sha256=SA78rt-FydtKQb4wiol75g0LpkRnqcKD8g7X_h24uSE,569
543
542
  polyaxon/_sdk/schemas/v1_service_account.py,sha256=03S_33y9_TBP-SCorFtwP6rxKYiFJxJyMPe32DIhXzg,569
544
- polyaxon/_sdk/schemas/v1_settings_catalog.py,sha256=Ee53jUnDqIZqbnQlIKmzg3yLurxGKl_PWue6QMdF-vw,340
543
+ polyaxon/_sdk/schemas/v1_settings_catalog.py,sha256=6EHl3U_m2Mnw0zTJkqRoD54T9ZzpqBvY8KEjmn-OIhM,311
545
544
  polyaxon/_sdk/schemas/v1_tag.py,sha256=dLAThV2aJbaQjE52w4rfGVmD5mXJqUN0jcs-LLDFfrk,411
546
- polyaxon/_sdk/schemas/v1_team.py,sha256=CUD3oHdjy00LoDFA9NtaKsM2qWrOksrz6VYVFYwdq-o,711
545
+ polyaxon/_sdk/schemas/v1_team.py,sha256=otrBvvYX92ZCLc2zaTtHBpKR3npOj80m2e6jSG_GLW0,679
547
546
  polyaxon/_sdk/schemas/v1_team_member.py,sha256=Sfj3-GH6iwf5auTbj9u3Uy6NGgYY5HP9n7kfkFrNtFw,450
548
547
  polyaxon/_sdk/schemas/v1_team_settings.py,sha256=bfWhOItuTCOnQQ15jkaYQWH4kwBKJI8QkwKj4vJ9xnw,301
549
548
  polyaxon/_sdk/schemas/v1_token.py,sha256=g2g2bfPtgBcZskfou-RO5sfDP9FqEoXYBgxoOkIhjC8,604
@@ -573,7 +572,7 @@ polyaxon/_utils/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
573
572
  polyaxon/_utils/cache.py,sha256=hsnADZUnnYKQ2jrzuQjg3z9_4YGznL-pof9wH-cObR8,1717
574
573
  polyaxon/_utils/cli_constants.py,sha256=v8qmVMEFYtYI33Fghsq5IlY7ceYgWa7osZB9d6NwlP8,795
575
574
  polyaxon/_utils/formatting.py,sha256=OLMfC7YRJKmmbF37_6xHr6tQKOdhEIU6Ij87h_tRDV0,2483
576
- polyaxon/_utils/fqn_utils.py,sha256=ixfFZyGdxvnuhaDbIJ6vTig2K48Z59ZqFMzisVkaDls,3260
575
+ polyaxon/_utils/fqn_utils.py,sha256=CX_CYn-kvun2cFNlpY9Gor007pKQS_yAqNta6RFujD0,2631
577
576
  polyaxon/_utils/host_utils.py,sha256=SAs_cC409rDO6qwUequOoybNUJw0fpNgOhljMiDlSdc,265
578
577
  polyaxon/_utils/test_utils.py,sha256=a6JU1T635ZseZWjHYx9Ga59fUcCBwfpFXy86Qo4fpMk,4873
579
578
  polyaxon/_utils/urls_utils.py,sha256=zdhR1M9r9kHcShT8GhvZaBsMi9mdKh-3wFSsvBPiw8c,1722
@@ -611,9 +610,9 @@ polyaxon/tuners/hyperopt.py,sha256=zd6MblMGkooqLGDFJVo5kClqYnBoMwGj-opqqj8FDzQ,7
611
610
  polyaxon/tuners/mapping.py,sha256=pOdHCiwEufTk-QT7pNyjBjAEWNTM-lMC17WNTCk7C24,69
612
611
  polyaxon/tuners/random_search.py,sha256=6VEekM3N9h6E1lbpVTTUGKFPJlGMY2u-GkG615_nQcI,80
613
612
  polyaxon_sdk/__init__.py,sha256=HWvFdGWESyVG3f26K_szewiG-McMOHFkXKTfZcBlHsM,92
614
- polyaxon-2.2.0.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
615
- polyaxon-2.2.0.dist-info/METADATA,sha256=vJfUFLkCTTcszj5-xxHxXAkRt8khaghUm5GEDKqij8I,11709
616
- polyaxon-2.2.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
617
- polyaxon-2.2.0.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
618
- polyaxon-2.2.0.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
619
- polyaxon-2.2.0.dist-info/RECORD,,
613
+ polyaxon-2.2.0rc0.dist-info/LICENSE,sha256=86kroZbQUDsmSWOomB7dpceG65UXiVSPob4581tStBc,11349
614
+ polyaxon-2.2.0rc0.dist-info/METADATA,sha256=6HGowcBr3B8VU_jXLLRAOuG3wC-qALrzwCvSfmM6tx4,11712
615
+ polyaxon-2.2.0rc0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
616
+ polyaxon-2.2.0rc0.dist-info/entry_points.txt,sha256=aFbUMjkg9vzRBVAFhqvR1m92yG8Cov7UAF0zViGfoQw,70
617
+ polyaxon-2.2.0rc0.dist-info/top_level.txt,sha256=I_2e_Vv8rdcqWcMMdZocbrHiKPNGqoSMBqIObrw00Rg,22
618
+ polyaxon-2.2.0rc0.dist-info/RECORD,,
polyaxon/_client/mixin.py DELETED
@@ -1,39 +0,0 @@
1
- from polyaxon import settings
2
- from polyaxon._client.client import PolyaxonClient
3
- from polyaxon._schemas.client import ClientConfig
4
- from polyaxon._utils.fqn_utils import split_owner_team_space
5
-
6
-
7
- class ClientMixin:
8
- @property
9
- def client(self):
10
- if self._client:
11
- return self._client
12
- self._client = PolyaxonClient()
13
- return self._client
14
-
15
- def reset_client(self, **kwargs):
16
- if not settings.CLIENT_CONFIG.in_cluster:
17
- self._client = PolyaxonClient(
18
- ClientConfig.patch_from(settings.CLIENT_CONFIG, **kwargs)
19
- )
20
-
21
- @property
22
- def owner(self) -> str:
23
- return self._owner
24
-
25
- @property
26
- def team(self):
27
- return self._team
28
-
29
- @property
30
- def project(self) -> str:
31
- return self._project
32
-
33
- def set_project(self, project: str):
34
- self._project = project
35
-
36
- def set_owner(self, owner: str):
37
- owner, team = split_owner_team_space(owner)
38
- self._owner = owner
39
- self._team = team