polyaxon 2.4.0rc1__py3-none-any.whl → 2.6.0__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.
Files changed (218) hide show
  1. polyaxon/_auxiliaries/default_scheduling.py +17 -7
  2. polyaxon/_auxiliaries/init.py +14 -6
  3. polyaxon/_auxiliaries/sidecar.py +10 -8
  4. polyaxon/_cli/dashboard.py +2 -5
  5. polyaxon/_cli/run.py +14 -0
  6. polyaxon/_cli/version.py +4 -1
  7. polyaxon/_compiler/contexts/contexts.py +4 -0
  8. polyaxon/_compiler/resolver/agent.py +10 -9
  9. polyaxon/_compiler/resolver/runtime.py +4 -0
  10. polyaxon/_constants/metadata.py +1 -0
  11. polyaxon/_contexts/keys.py +1 -0
  12. polyaxon/_deploy/schemas/auth.py +3 -3
  13. polyaxon/_deploy/schemas/celery.py +10 -8
  14. polyaxon/_deploy/schemas/deployment.py +148 -116
  15. polyaxon/_deploy/schemas/email.py +8 -8
  16. polyaxon/_deploy/schemas/ingress.py +7 -7
  17. polyaxon/_deploy/schemas/intervals.py +3 -1
  18. polyaxon/_deploy/schemas/operators.py +8 -8
  19. polyaxon/_deploy/schemas/proxy.py +9 -9
  20. polyaxon/_deploy/schemas/rbac.py +1 -1
  21. polyaxon/_deploy/schemas/root_user.py +5 -5
  22. polyaxon/_deploy/schemas/security_context.py +25 -15
  23. polyaxon/_deploy/schemas/service.py +73 -69
  24. polyaxon/_deploy/schemas/ssl.py +3 -3
  25. polyaxon/_deploy/schemas/ui.py +10 -6
  26. polyaxon/_docker/builder/builder.py +4 -1
  27. polyaxon/_docker/converter/base/containers.py +4 -7
  28. polyaxon/_docker/converter/base/env_vars.py +5 -5
  29. polyaxon/_docker/converter/base/mounts.py +2 -2
  30. polyaxon/_docker/docker_types.py +57 -30
  31. polyaxon/_env_vars/keys.py +2 -0
  32. polyaxon/_flow/__init__.py +3 -2
  33. polyaxon/_flow/builds/__init__.py +8 -8
  34. polyaxon/_flow/cache/__init__.py +4 -4
  35. polyaxon/_flow/component/base.py +25 -18
  36. polyaxon/_flow/component/component.py +4 -3
  37. polyaxon/_flow/early_stopping/__init__.py +1 -1
  38. polyaxon/_flow/early_stopping/policies.py +12 -10
  39. polyaxon/_flow/environment/__init__.py +42 -24
  40. polyaxon/_flow/events/__init__.py +1 -1
  41. polyaxon/_flow/hooks/__init__.py +11 -11
  42. polyaxon/_flow/init/__init__.py +41 -25
  43. polyaxon/_flow/io/io.py +57 -47
  44. polyaxon/_flow/joins/__init__.py +5 -5
  45. polyaxon/_flow/matrix/bayes.py +23 -17
  46. polyaxon/_flow/matrix/grid_search.py +16 -7
  47. polyaxon/_flow/matrix/hyperband.py +10 -10
  48. polyaxon/_flow/matrix/hyperopt.py +14 -9
  49. polyaxon/_flow/matrix/iterative.py +14 -8
  50. polyaxon/_flow/matrix/mapping.py +4 -4
  51. polyaxon/_flow/matrix/params.py +138 -77
  52. polyaxon/_flow/matrix/random_search.py +10 -5
  53. polyaxon/_flow/matrix/tuner.py +4 -4
  54. polyaxon/_flow/mounts/artifacts_mounts.py +1 -1
  55. polyaxon/_flow/notifications/__init__.py +1 -1
  56. polyaxon/_flow/operations/base.py +10 -8
  57. polyaxon/_flow/operations/compiled_operation.py +5 -4
  58. polyaxon/_flow/operations/operation.py +30 -22
  59. polyaxon/_flow/optimization/__init__.py +2 -2
  60. polyaxon/_flow/params/params.py +10 -9
  61. polyaxon/_flow/plugins/__init__.py +19 -13
  62. polyaxon/_flow/run/dag.py +12 -9
  63. polyaxon/_flow/run/dask/dask.py +4 -4
  64. polyaxon/_flow/run/dask/replica.py +17 -11
  65. polyaxon/_flow/run/job.py +17 -11
  66. polyaxon/_flow/run/kubeflow/mpi_job.py +10 -5
  67. polyaxon/_flow/run/kubeflow/mx_job.py +25 -9
  68. polyaxon/_flow/run/kubeflow/paddle_job.py +16 -9
  69. polyaxon/_flow/run/kubeflow/pytorch_job.py +24 -17
  70. polyaxon/_flow/run/kubeflow/replica.py +17 -11
  71. polyaxon/_flow/run/kubeflow/scheduling_policy.py +7 -5
  72. polyaxon/_flow/run/kubeflow/tf_job.py +15 -8
  73. polyaxon/_flow/run/kubeflow/xgboost_job.py +9 -4
  74. polyaxon/_flow/run/ray/ray.py +9 -6
  75. polyaxon/_flow/run/ray/replica.py +25 -16
  76. polyaxon/_flow/run/resources.py +14 -13
  77. polyaxon/_flow/run/service.py +4 -4
  78. polyaxon/_flow/schedules/cron.py +4 -4
  79. polyaxon/_flow/schedules/interval.py +4 -4
  80. polyaxon/_flow/templates/__init__.py +3 -3
  81. polyaxon/_flow/termination/__init__.py +3 -3
  82. polyaxon/_fs/async_manager.py +1 -1
  83. polyaxon/_fs/watcher.py +26 -27
  84. polyaxon/_k8s/k8s_validation.py +1 -1
  85. polyaxon/_k8s/logging/async_monitor.py +18 -3
  86. polyaxon/_local_process/converter/base/containers.py +4 -7
  87. polyaxon/_local_process/converter/base/env_vars.py +5 -5
  88. polyaxon/_local_process/process_types.py +15 -12
  89. polyaxon/_polyaxonfile/specs/compiled_operation.py +1 -1
  90. polyaxon/_polyaxonfile/specs/libs/parser.py +1 -1
  91. polyaxon/_polyaxonfile/specs/libs/validator.py +1 -1
  92. polyaxon/_polyaxonfile/specs/operation.py +1 -1
  93. polyaxon/_polyaxonfile/specs/sections.py +8 -0
  94. polyaxon/_runner/agent/async_agent.py +9 -6
  95. polyaxon/_runner/agent/base_agent.py +8 -5
  96. polyaxon/_runner/agent/sync_agent.py +8 -5
  97. polyaxon/_runner/converter/converter.py +12 -4
  98. polyaxon/_schemas/agent.py +69 -37
  99. polyaxon/_schemas/authentication.py +4 -4
  100. polyaxon/_schemas/base.py +26 -2
  101. polyaxon/_schemas/checks.py +3 -3
  102. polyaxon/_schemas/cli.py +4 -6
  103. polyaxon/_schemas/client.py +20 -18
  104. polyaxon/_schemas/compatibility.py +4 -4
  105. polyaxon/_schemas/container_resources.py +1 -1
  106. polyaxon/_schemas/home.py +3 -3
  107. polyaxon/_schemas/installation.py +13 -9
  108. polyaxon/_schemas/lifecycle.py +23 -23
  109. polyaxon/_schemas/log_handler.py +2 -2
  110. polyaxon/_schemas/services.py +26 -14
  111. polyaxon/_schemas/types/artifacts.py +3 -3
  112. polyaxon/_schemas/types/dockerfile.py +14 -12
  113. polyaxon/_schemas/types/event.py +2 -2
  114. polyaxon/_schemas/types/file.py +3 -3
  115. polyaxon/_schemas/types/git.py +12 -4
  116. polyaxon/_schemas/types/tensorboard.py +14 -8
  117. polyaxon/_schemas/user.py +3 -3
  118. polyaxon/_schemas/version.py +2 -2
  119. polyaxon/_sdk/api/agents_v1_api.py +45 -45
  120. polyaxon/_sdk/api/artifacts_stores_v1_api.py +3 -3
  121. polyaxon/_sdk/api/auth_v1_api.py +13 -13
  122. polyaxon/_sdk/api/connections_v1_api.py +15 -15
  123. polyaxon/_sdk/api/dashboards_v1_api.py +15 -15
  124. polyaxon/_sdk/api/organizations_v1_api.py +77 -77
  125. polyaxon/_sdk/api/presets_v1_api.py +15 -15
  126. polyaxon/_sdk/api/project_dashboards_v1_api.py +17 -17
  127. polyaxon/_sdk/api/project_searches_v1_api.py +17 -17
  128. polyaxon/_sdk/api/projects_v1_api.py +65 -65
  129. polyaxon/_sdk/api/queues_v1_api.py +19 -19
  130. polyaxon/_sdk/api/runs_v1_api.py +127 -127
  131. polyaxon/_sdk/api/searches_v1_api.py +15 -15
  132. polyaxon/_sdk/api/service_accounts_v1_api.py +27 -27
  133. polyaxon/_sdk/api/tags_v1_api.py +17 -17
  134. polyaxon/_sdk/api/teams_v1_api.py +51 -51
  135. polyaxon/_sdk/api/users_v1_api.py +25 -25
  136. polyaxon/_sdk/api/versions_v1_api.py +7 -7
  137. polyaxon/_sdk/schemas/v1_activity.py +8 -8
  138. polyaxon/_sdk/schemas/v1_agent.py +17 -16
  139. polyaxon/_sdk/schemas/v1_agent_state_response.py +4 -4
  140. polyaxon/_sdk/schemas/v1_agent_state_response_agent_state.py +10 -10
  141. polyaxon/_sdk/schemas/v1_agent_status_body_request.py +3 -3
  142. polyaxon/_sdk/schemas/v1_analytics_spec.py +4 -4
  143. polyaxon/_sdk/schemas/v1_artifact_tree.py +3 -3
  144. polyaxon/_sdk/schemas/v1_auth.py +1 -1
  145. polyaxon/_sdk/schemas/v1_cloning.py +3 -3
  146. polyaxon/_sdk/schemas/v1_connection_response.py +9 -9
  147. polyaxon/_sdk/schemas/v1_dashboard.py +9 -9
  148. polyaxon/_sdk/schemas/v1_dashboard_spec.py +5 -5
  149. polyaxon/_sdk/schemas/v1_entities_tags.py +2 -2
  150. polyaxon/_sdk/schemas/v1_entities_transfer.py +2 -2
  151. polyaxon/_sdk/schemas/v1_entity_notification_body.py +7 -7
  152. polyaxon/_sdk/schemas/v1_entity_stage_body_request.py +5 -5
  153. polyaxon/_sdk/schemas/v1_entity_status_body_request.py +5 -5
  154. polyaxon/_sdk/schemas/v1_events_response.py +2 -2
  155. polyaxon/_sdk/schemas/v1_list_activities_response.py +4 -4
  156. polyaxon/_sdk/schemas/v1_list_agents_response.py +4 -4
  157. polyaxon/_sdk/schemas/v1_list_bookmarks_response.py +4 -4
  158. polyaxon/_sdk/schemas/v1_list_connections_response.py +4 -4
  159. polyaxon/_sdk/schemas/v1_list_dashboards_response.py +4 -4
  160. polyaxon/_sdk/schemas/v1_list_organization_members_response.py +4 -4
  161. polyaxon/_sdk/schemas/v1_list_organizations_response.py +4 -4
  162. polyaxon/_sdk/schemas/v1_list_presets_response.py +4 -4
  163. polyaxon/_sdk/schemas/v1_list_project_versions_response.py +4 -4
  164. polyaxon/_sdk/schemas/v1_list_projects_response.py +4 -4
  165. polyaxon/_sdk/schemas/v1_list_queues_response.py +4 -4
  166. polyaxon/_sdk/schemas/v1_list_run_artifacts_response.py +4 -4
  167. polyaxon/_sdk/schemas/v1_list_run_connections_response.py +4 -4
  168. polyaxon/_sdk/schemas/v1_list_run_edges_response.py +4 -4
  169. polyaxon/_sdk/schemas/v1_list_runs_response.py +4 -4
  170. polyaxon/_sdk/schemas/v1_list_searches_response.py +4 -4
  171. polyaxon/_sdk/schemas/v1_list_service_accounts_response.py +4 -4
  172. polyaxon/_sdk/schemas/v1_list_tags_response.py +4 -4
  173. polyaxon/_sdk/schemas/v1_list_team_members_response.py +4 -4
  174. polyaxon/_sdk/schemas/v1_list_teams_response.py +4 -4
  175. polyaxon/_sdk/schemas/v1_list_token_response.py +4 -4
  176. polyaxon/_sdk/schemas/v1_operation_body.py +8 -8
  177. polyaxon/_sdk/schemas/v1_organization.py +16 -16
  178. polyaxon/_sdk/schemas/v1_organization_member.py +6 -6
  179. polyaxon/_sdk/schemas/v1_password_change.py +3 -3
  180. polyaxon/_sdk/schemas/v1_pipeline.py +3 -3
  181. polyaxon/_sdk/schemas/v1_preset.py +11 -12
  182. polyaxon/_sdk/schemas/v1_project.py +17 -17
  183. polyaxon/_sdk/schemas/v1_project_settings.py +11 -11
  184. polyaxon/_sdk/schemas/v1_project_version.py +20 -20
  185. polyaxon/_sdk/schemas/v1_queue.py +12 -12
  186. polyaxon/_sdk/schemas/v1_run.py +38 -38
  187. polyaxon/_sdk/schemas/v1_run_connection.py +3 -3
  188. polyaxon/_sdk/schemas/v1_run_edge.py +5 -5
  189. polyaxon/_sdk/schemas/v1_run_edge_lineage.py +3 -3
  190. polyaxon/_sdk/schemas/v1_run_edges_graph.py +1 -1
  191. polyaxon/_sdk/schemas/v1_run_reference_catalog.py +4 -4
  192. polyaxon/_sdk/schemas/v1_run_settings.py +9 -9
  193. polyaxon/_sdk/schemas/v1_search.py +10 -10
  194. polyaxon/_sdk/schemas/v1_search_spec.py +14 -14
  195. polyaxon/_sdk/schemas/v1_section_spec.py +12 -12
  196. polyaxon/_sdk/schemas/v1_service_account.py +9 -9
  197. polyaxon/_sdk/schemas/v1_settings_catalog.py +4 -4
  198. polyaxon/_sdk/schemas/v1_tag.py +6 -6
  199. polyaxon/_sdk/schemas/v1_team.py +11 -11
  200. polyaxon/_sdk/schemas/v1_team_member.py +6 -6
  201. polyaxon/_sdk/schemas/v1_team_settings.py +2 -2
  202. polyaxon/_sdk/schemas/v1_token.py +10 -10
  203. polyaxon/_sdk/schemas/v1_trial_start.py +6 -6
  204. polyaxon/_sdk/schemas/v1_user.py +6 -6
  205. polyaxon/_sdk/schemas/v1_user_access.py +7 -7
  206. polyaxon/_sdk/schemas/v1_user_email.py +1 -1
  207. polyaxon/_sdk/schemas/v1_user_singup.py +5 -5
  208. polyaxon/_sdk/schemas/v1_uuids.py +1 -1
  209. polyaxon/_sidecar/container/__init__.py +13 -8
  210. polyaxon/_utils/cli_constants.py +2 -0
  211. polyaxon/_utils/test_utils.py +2 -1
  212. polyaxon/pkg.py +1 -1
  213. {polyaxon-2.4.0rc1.dist-info → polyaxon-2.6.0.dist-info}/METADATA +13 -13
  214. {polyaxon-2.4.0rc1.dist-info → polyaxon-2.6.0.dist-info}/RECORD +218 -218
  215. {polyaxon-2.4.0rc1.dist-info → polyaxon-2.6.0.dist-info}/WHEEL +1 -1
  216. {polyaxon-2.4.0rc1.dist-info → polyaxon-2.6.0.dist-info}/LICENSE +0 -0
  217. {polyaxon-2.4.0rc1.dist-info → polyaxon-2.6.0.dist-info}/entry_points.txt +0 -0
  218. {polyaxon-2.4.0rc1.dist-info → polyaxon-2.6.0.dist-info}/top_level.txt +0 -0
@@ -1,7 +1,7 @@
1
1
  from typing import Optional
2
2
  from typing_extensions import Annotated
3
3
 
4
- from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_arguments
4
+ from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_call
5
5
 
6
6
  from polyaxon._sdk.base_api import BaseApi
7
7
  from polyaxon._sdk.schemas.v1_list_queues_response import V1ListQueuesResponse
@@ -10,7 +10,7 @@ from polyaxon.exceptions import ApiTypeError
10
10
 
11
11
 
12
12
  class QueuesV1Api(BaseApi):
13
- @validate_arguments
13
+ @validate_call
14
14
  def create_queue(
15
15
  self,
16
16
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -54,7 +54,7 @@ class QueuesV1Api(BaseApi):
54
54
  owner, agent, body, **kwargs
55
55
  ) # noqa: E501
56
56
 
57
- @validate_arguments
57
+ @validate_call
58
58
  def create_queue_with_http_info(
59
59
  self,
60
60
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -193,7 +193,7 @@ class QueuesV1Api(BaseApi):
193
193
  _request_auth=_params.get("_request_auth"),
194
194
  )
195
195
 
196
- @validate_arguments
196
+ @validate_call
197
197
  def delete_queue(
198
198
  self,
199
199
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -242,7 +242,7 @@ class QueuesV1Api(BaseApi):
242
242
  owner, entity, uuid, **kwargs
243
243
  ) # noqa: E501
244
244
 
245
- @validate_arguments
245
+ @validate_call
246
246
  def delete_queue_with_http_info(
247
247
  self,
248
248
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -373,7 +373,7 @@ class QueuesV1Api(BaseApi):
373
373
  _request_auth=_params.get("_request_auth"),
374
374
  )
375
375
 
376
- @validate_arguments
376
+ @validate_call
377
377
  def get_queue(
378
378
  self,
379
379
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -422,7 +422,7 @@ class QueuesV1Api(BaseApi):
422
422
  owner, entity, uuid, **kwargs
423
423
  ) # noqa: E501
424
424
 
425
- @validate_arguments
425
+ @validate_call
426
426
  def get_queue_with_http_info(
427
427
  self,
428
428
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -558,7 +558,7 @@ class QueuesV1Api(BaseApi):
558
558
  _request_auth=_params.get("_request_auth"),
559
559
  )
560
560
 
561
- @validate_arguments
561
+ @validate_call
562
562
  def list_organization_queue_names(
563
563
  self,
564
564
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -625,7 +625,7 @@ class QueuesV1Api(BaseApi):
625
625
  owner, offset, limit, sort, query, bookmarks, mode, no_page, **kwargs
626
626
  ) # noqa: E501
627
627
 
628
- @validate_arguments
628
+ @validate_call
629
629
  def list_organization_queue_names_with_http_info(
630
630
  self,
631
631
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -798,7 +798,7 @@ class QueuesV1Api(BaseApi):
798
798
  _request_auth=_params.get("_request_auth"),
799
799
  )
800
800
 
801
- @validate_arguments
801
+ @validate_call
802
802
  def list_organization_queues(
803
803
  self,
804
804
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -865,7 +865,7 @@ class QueuesV1Api(BaseApi):
865
865
  owner, offset, limit, sort, query, bookmarks, mode, no_page, **kwargs
866
866
  ) # noqa: E501
867
867
 
868
- @validate_arguments
868
+ @validate_call
869
869
  def list_organization_queues_with_http_info(
870
870
  self,
871
871
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1038,7 +1038,7 @@ class QueuesV1Api(BaseApi):
1038
1038
  _request_auth=_params.get("_request_auth"),
1039
1039
  )
1040
1040
 
1041
- @validate_arguments
1041
+ @validate_call
1042
1042
  def list_queue_names(
1043
1043
  self,
1044
1044
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1110,7 +1110,7 @@ class QueuesV1Api(BaseApi):
1110
1110
  owner, name, offset, limit, sort, query, bookmarks, mode, no_page, **kwargs
1111
1111
  ) # noqa: E501
1112
1112
 
1113
- @validate_arguments
1113
+ @validate_call
1114
1114
  def list_queue_names_with_http_info(
1115
1115
  self,
1116
1116
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1291,7 +1291,7 @@ class QueuesV1Api(BaseApi):
1291
1291
  _request_auth=_params.get("_request_auth"),
1292
1292
  )
1293
1293
 
1294
- @validate_arguments
1294
+ @validate_call
1295
1295
  def list_queues(
1296
1296
  self,
1297
1297
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1363,7 +1363,7 @@ class QueuesV1Api(BaseApi):
1363
1363
  owner, name, offset, limit, sort, query, bookmarks, mode, no_page, **kwargs
1364
1364
  ) # noqa: E501
1365
1365
 
1366
- @validate_arguments
1366
+ @validate_call
1367
1367
  def list_queues_with_http_info(
1368
1368
  self,
1369
1369
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1544,7 +1544,7 @@ class QueuesV1Api(BaseApi):
1544
1544
  _request_auth=_params.get("_request_auth"),
1545
1545
  )
1546
1546
 
1547
- @validate_arguments
1547
+ @validate_call
1548
1548
  def patch_queue(
1549
1549
  self,
1550
1550
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1591,7 +1591,7 @@ class QueuesV1Api(BaseApi):
1591
1591
  owner, agent, queue_uuid, body, **kwargs
1592
1592
  ) # noqa: E501
1593
1593
 
1594
- @validate_arguments
1594
+ @validate_call
1595
1595
  def patch_queue_with_http_info(
1596
1596
  self,
1597
1597
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1735,7 +1735,7 @@ class QueuesV1Api(BaseApi):
1735
1735
  _request_auth=_params.get("_request_auth"),
1736
1736
  )
1737
1737
 
1738
- @validate_arguments
1738
+ @validate_call
1739
1739
  def update_queue(
1740
1740
  self,
1741
1741
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
@@ -1782,7 +1782,7 @@ class QueuesV1Api(BaseApi):
1782
1782
  owner, agent, queue_uuid, body, **kwargs
1783
1783
  ) # noqa: E501
1784
1784
 
1785
- @validate_arguments
1785
+ @validate_call
1786
1786
  def update_queue_with_http_info(
1787
1787
  self,
1788
1788
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],