polyaxon-sdk 2.1.0rc3__py2.py3-none-any.whl → 2.1.0rc5__py2.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_sdk/__init__.py +4 -2
- polyaxon_sdk/api/agents_v1_api.py +1 -1
- polyaxon_sdk/api/artifacts_stores_v1_api.py +1 -1
- polyaxon_sdk/api/auth_v1_api.py +1 -1
- polyaxon_sdk/api/connections_v1_api.py +1 -1
- polyaxon_sdk/api/dashboards_v1_api.py +1 -1
- polyaxon_sdk/api/organizations_v1_api.py +1 -1
- polyaxon_sdk/api/presets_v1_api.py +1 -1
- polyaxon_sdk/api/project_dashboards_v1_api.py +1 -1
- polyaxon_sdk/api/project_searches_v1_api.py +1 -1
- polyaxon_sdk/api/projects_v1_api.py +1 -1
- polyaxon_sdk/api/queues_v1_api.py +1 -1
- polyaxon_sdk/api/runs_v1_api.py +1 -1
- polyaxon_sdk/api/schemas_v1_api.py +1 -1
- polyaxon_sdk/api/searches_v1_api.py +1 -1
- polyaxon_sdk/api/service_accounts_v1_api.py +1 -1
- polyaxon_sdk/api/tags_v1_api.py +1 -1
- polyaxon_sdk/api/teams_v1_api.py +1 -1
- polyaxon_sdk/api/users_v1_api.py +1 -1
- polyaxon_sdk/api/versions_v1_api.py +1 -1
- polyaxon_sdk/api_client.py +2 -2
- polyaxon_sdk/configuration.py +3 -3
- polyaxon_sdk/exceptions.py +1 -1
- polyaxon_sdk/models/__init__.py +3 -1
- polyaxon_sdk/models/agent_state_response_agent_state.py +1 -1
- polyaxon_sdk/models/mx_job_mode.py +1 -1
- polyaxon_sdk/models/protobuf_any.py +1 -1
- polyaxon_sdk/models/protobuf_null_value.py +1 -1
- polyaxon_sdk/models/runtime_error.py +1 -1
- polyaxon_sdk/models/search_view.py +1 -1
- polyaxon_sdk/models/v1_activity.py +1 -1
- polyaxon_sdk/models/v1_agent.py +1 -1
- polyaxon_sdk/models/v1_agent_state_response.py +1 -1
- polyaxon_sdk/models/v1_agent_status_body_request.py +1 -1
- polyaxon_sdk/models/v1_analytics_spec.py +1 -1
- polyaxon_sdk/models/v1_artifact_kind.py +1 -1
- polyaxon_sdk/models/v1_artifact_tree.py +1 -1
- polyaxon_sdk/models/v1_artifacts_mount.py +1 -1
- polyaxon_sdk/models/v1_artifacts_type.py +1 -1
- polyaxon_sdk/models/v1_auth.py +1 -1
- polyaxon_sdk/models/v1_auth_type.py +1 -1
- polyaxon_sdk/models/v1_average_stopping_policy.py +1 -1
- polyaxon_sdk/models/v1_bayes.py +1 -1
- polyaxon_sdk/models/v1_bucket_connection.py +1 -1
- polyaxon_sdk/models/v1_build.py +1 -1
- polyaxon_sdk/models/v1_cache.py +1 -1
- polyaxon_sdk/models/v1_claim_connection.py +1 -1
- polyaxon_sdk/models/v1_clean_pod_policy.py +1 -1
- polyaxon_sdk/models/v1_cloning.py +1 -1
- polyaxon_sdk/models/v1_cloning_kind.py +1 -1
- polyaxon_sdk/models/v1_compatibility.py +1 -1
- polyaxon_sdk/models/v1_compiled_operation.py +1 -1
- polyaxon_sdk/models/v1_component.py +1 -1
- polyaxon_sdk/models/v1_connection_kind.py +1 -1
- polyaxon_sdk/models/v1_connection_resource.py +1 -1
- polyaxon_sdk/models/v1_connection_response.py +1 -1
- polyaxon_sdk/models/v1_connection_schema.py +1 -1
- polyaxon_sdk/models/v1_connection_type.py +1 -1
- polyaxon_sdk/models/v1_credentials.py +1 -1
- polyaxon_sdk/models/v1_cron_schedule.py +1 -1
- polyaxon_sdk/models/v1_dag.py +1 -1
- polyaxon_sdk/models/v1_dag_ref.py +1 -1
- polyaxon_sdk/models/v1_dashboard.py +1 -1
- polyaxon_sdk/models/v1_dashboard_spec.py +1 -1
- polyaxon_sdk/models/v1_dask_job.py +1 -1
- polyaxon_sdk/models/v1_dask_replica.py +1 -1
- polyaxon_sdk/models/v1_date_time_schedule.py +1 -1
- polyaxon_sdk/models/v1_diff_stopping_policy.py +1 -1
- polyaxon_sdk/models/v1_dockerfile_type.py +1 -1
- polyaxon_sdk/models/v1_early_stopping.py +1 -1
- polyaxon_sdk/models/v1_entities_tags.py +1 -1
- polyaxon_sdk/models/v1_entities_transfer.py +1 -1
- polyaxon_sdk/models/v1_entity_notification_body.py +1 -1
- polyaxon_sdk/models/v1_entity_stage_body_request.py +1 -1
- polyaxon_sdk/models/v1_entity_status_body_request.py +1 -1
- polyaxon_sdk/models/v1_environment.py +1 -1
- polyaxon_sdk/models/v1_event.py +1 -1
- polyaxon_sdk/models/v1_event_artifact.py +1 -1
- polyaxon_sdk/models/v1_event_audio.py +1 -1
- polyaxon_sdk/models/v1_event_chart.py +1 -1
- polyaxon_sdk/models/v1_event_chart_kind.py +1 -1
- polyaxon_sdk/models/v1_event_confusion_matrix.py +1 -1
- polyaxon_sdk/models/v1_event_curve.py +1 -1
- polyaxon_sdk/models/v1_event_curve_kind.py +1 -1
- polyaxon_sdk/models/v1_event_dataframe.py +1 -1
- polyaxon_sdk/models/v1_event_histogram.py +1 -1
- polyaxon_sdk/models/v1_event_image.py +1 -1
- polyaxon_sdk/models/v1_event_kind.py +1 -1
- polyaxon_sdk/models/v1_event_model.py +1 -1
- polyaxon_sdk/models/v1_event_span.py +1 -1
- polyaxon_sdk/models/v1_event_span_kind.py +1 -1
- polyaxon_sdk/models/v1_event_trigger.py +1 -1
- polyaxon_sdk/models/v1_event_type.py +1 -1
- polyaxon_sdk/models/v1_event_video.py +1 -1
- polyaxon_sdk/models/v1_events_response.py +1 -1
- polyaxon_sdk/models/v1_failure_early_stopping.py +1 -1
- polyaxon_sdk/models/v1_file_type.py +1 -1
- polyaxon_sdk/models/v1_gcs_type.py +1 -1
- polyaxon_sdk/models/v1_git_connection.py +1 -1
- polyaxon_sdk/models/v1_git_type.py +1 -1
- polyaxon_sdk/models/v1_grid_search.py +1 -1
- polyaxon_sdk/models/v1_hook.py +1 -1
- polyaxon_sdk/models/v1_host_connection.py +1 -1
- polyaxon_sdk/models/v1_host_path_connection.py +1 -1
- polyaxon_sdk/models/v1_hp_choice.py +1 -1
- polyaxon_sdk/models/v1_hp_date_range.py +1 -1
- polyaxon_sdk/models/v1_hp_date_time_range.py +1 -1
- polyaxon_sdk/models/v1_hp_geom_space.py +1 -1
- polyaxon_sdk/models/v1_hp_lin_space.py +1 -1
- polyaxon_sdk/models/v1_hp_log_normal.py +1 -1
- polyaxon_sdk/models/v1_hp_log_space.py +1 -1
- polyaxon_sdk/models/v1_hp_log_uniform.py +1 -1
- polyaxon_sdk/models/v1_hp_normal.py +1 -1
- polyaxon_sdk/models/v1_hp_p_choice.py +1 -1
- polyaxon_sdk/models/v1_hp_params.py +1 -1
- polyaxon_sdk/models/v1_hp_q_log_normal.py +1 -1
- polyaxon_sdk/models/v1_hp_q_log_uniform.py +1 -1
- polyaxon_sdk/models/v1_hp_q_normal.py +1 -1
- polyaxon_sdk/models/v1_hp_q_uniform.py +1 -1
- polyaxon_sdk/models/v1_hp_range.py +1 -1
- polyaxon_sdk/models/v1_hp_uniform.py +1 -1
- polyaxon_sdk/models/v1_hub_ref.py +1 -1
- polyaxon_sdk/models/v1_hyperband.py +1 -1
- polyaxon_sdk/models/v1_hyperopt.py +1 -1
- polyaxon_sdk/models/v1_hyperopt_algorithms.py +1 -1
- polyaxon_sdk/models/v1_init.py +1 -1
- polyaxon_sdk/models/v1_installation.py +1 -1
- polyaxon_sdk/models/v1_interval_schedule.py +1 -1
- polyaxon_sdk/models/v1_io.py +1 -1
- polyaxon_sdk/models/v1_iterative.py +1 -1
- polyaxon_sdk/models/v1_job.py +1 -1
- polyaxon_sdk/models/v1_join.py +1 -1
- polyaxon_sdk/models/v1_join_param.py +1 -1
- polyaxon_sdk/models/v1_kf_replica.py +1 -1
- polyaxon_sdk/models/v1_list_activities_response.py +1 -1
- polyaxon_sdk/models/v1_list_agents_response.py +1 -1
- polyaxon_sdk/models/v1_list_bookmarks_response.py +1 -1
- polyaxon_sdk/models/v1_list_connections_response.py +1 -1
- polyaxon_sdk/models/v1_list_dashboards_response.py +1 -1
- polyaxon_sdk/models/v1_list_organization_members_response.py +1 -1
- polyaxon_sdk/models/v1_list_organizations_response.py +1 -1
- polyaxon_sdk/models/v1_list_presets_response.py +1 -1
- polyaxon_sdk/models/v1_list_project_versions_response.py +1 -1
- polyaxon_sdk/models/v1_list_projects_response.py +1 -1
- polyaxon_sdk/models/v1_list_queues_response.py +1 -1
- polyaxon_sdk/models/v1_list_run_artifacts_response.py +1 -1
- polyaxon_sdk/models/v1_list_run_connections_response.py +1 -1
- polyaxon_sdk/models/v1_list_run_edges_response.py +1 -1
- polyaxon_sdk/models/v1_list_runs_response.py +1 -1
- polyaxon_sdk/models/v1_list_searches_response.py +1 -1
- polyaxon_sdk/models/v1_list_service_accounts_response.py +1 -1
- polyaxon_sdk/models/v1_list_tags_response.py +1 -1
- polyaxon_sdk/models/v1_list_team_members_response.py +1 -1
- polyaxon_sdk/models/v1_list_teams_response.py +1 -1
- polyaxon_sdk/models/v1_list_token_response.py +1 -1
- polyaxon_sdk/models/v1_log.py +1 -1
- polyaxon_sdk/models/v1_log_handler.py +1 -1
- polyaxon_sdk/models/v1_logs.py +1 -1
- polyaxon_sdk/models/v1_managed_by.py +1 -1
- polyaxon_sdk/models/v1_mapping.py +1 -1
- polyaxon_sdk/models/v1_matrix.py +1 -1
- polyaxon_sdk/models/v1_matrix_kind.py +1 -1
- polyaxon_sdk/models/v1_median_stopping_policy.py +1 -1
- polyaxon_sdk/models/v1_metric_early_stopping.py +1 -1
- polyaxon_sdk/models/v1_mpi_job.py +1 -1
- polyaxon_sdk/models/v1_multi_events_response.py +1 -1
- polyaxon_sdk/models/v1_mx_job.py +1 -1
- polyaxon_sdk/models/v1_notification.py +1 -1
- polyaxon_sdk/models/v1_operation.py +1 -1
- polyaxon_sdk/models/v1_operation_body.py +1 -1
- polyaxon_sdk/models/v1_optimization.py +1 -1
- polyaxon_sdk/models/v1_optimization_metric.py +1 -1
- polyaxon_sdk/models/v1_optimization_resource.py +1 -1
- polyaxon_sdk/models/v1_organization.py +1 -1
- polyaxon_sdk/models/v1_organization_member.py +1 -1
- polyaxon_sdk/models/v1_paddle_elastic_polic.py +77 -0
- polyaxon_sdk/models/v1_paddle_job.py +8 -2
- polyaxon_sdk/models/v1_param.py +1 -1
- polyaxon_sdk/models/v1_password_change.py +1 -1
- polyaxon_sdk/models/v1_patch_strategy.py +1 -1
- polyaxon_sdk/models/v1_path_ref.py +1 -1
- polyaxon_sdk/models/v1_pipeline.py +1 -1
- polyaxon_sdk/models/v1_pipeline_kind.py +1 -1
- polyaxon_sdk/models/v1_plugins.py +1 -1
- polyaxon_sdk/models/v1_polyaxon_init_container.py +1 -1
- polyaxon_sdk/models/v1_polyaxon_sidecar_container.py +1 -1
- polyaxon_sdk/models/v1_preset.py +1 -1
- polyaxon_sdk/models/v1_project.py +1 -1
- polyaxon_sdk/models/v1_project_settings.py +1 -1
- polyaxon_sdk/models/v1_project_user_access.py +1 -1
- polyaxon_sdk/models/v1_project_version.py +1 -1
- polyaxon_sdk/models/v1_project_version_kind.py +1 -1
- polyaxon_sdk/models/v1_pytorch_elastic_policy.py +91 -0
- polyaxon_sdk/models/v1_pytorch_job.py +11 -3
- polyaxon_sdk/models/v1_queue.py +1 -1
- polyaxon_sdk/models/v1_random_search.py +1 -1
- polyaxon_sdk/models/v1_ray_job.py +1 -1
- polyaxon_sdk/models/v1_ray_replica.py +1 -1
- polyaxon_sdk/models/v1_reference.py +1 -1
- polyaxon_sdk/models/v1_resource_type.py +1 -1
- polyaxon_sdk/models/v1_run.py +1 -1
- polyaxon_sdk/models/v1_run_artifact.py +1 -1
- polyaxon_sdk/models/v1_run_artifacts.py +1 -1
- polyaxon_sdk/models/v1_run_connection.py +1 -1
- polyaxon_sdk/models/v1_run_edge.py +1 -1
- polyaxon_sdk/models/v1_run_edge_kind.py +1 -1
- polyaxon_sdk/models/v1_run_edge_lineage.py +1 -1
- polyaxon_sdk/models/v1_run_edges_graph.py +1 -1
- polyaxon_sdk/models/v1_run_kind.py +1 -1
- polyaxon_sdk/models/v1_run_pending.py +1 -1
- polyaxon_sdk/models/v1_run_reference_catalog.py +1 -1
- polyaxon_sdk/models/v1_run_resources.py +1 -1
- polyaxon_sdk/models/v1_run_schema.py +1 -1
- polyaxon_sdk/models/v1_run_settings.py +1 -1
- polyaxon_sdk/models/v1_s3_type.py +1 -1
- polyaxon_sdk/models/v1_schedule.py +1 -1
- polyaxon_sdk/models/v1_schedule_kind.py +1 -1
- polyaxon_sdk/models/v1_scheduling_policy.py +8 -4
- polyaxon_sdk/models/v1_schemas.py +1 -1
- polyaxon_sdk/models/v1_search.py +1 -1
- polyaxon_sdk/models/v1_search_spec.py +1 -1
- polyaxon_sdk/models/v1_section_spec.py +1 -1
- polyaxon_sdk/models/v1_service.py +1 -1
- polyaxon_sdk/models/v1_service_account.py +1 -1
- polyaxon_sdk/models/v1_settings_catalog.py +1 -1
- polyaxon_sdk/models/v1_stage.py +1 -1
- polyaxon_sdk/models/v1_stage_condition.py +1 -1
- polyaxon_sdk/models/v1_stages.py +1 -1
- polyaxon_sdk/models/v1_status.py +1 -1
- polyaxon_sdk/models/v1_status_condition.py +1 -1
- polyaxon_sdk/models/v1_statuses.py +1 -1
- polyaxon_sdk/models/v1_tag.py +1 -1
- polyaxon_sdk/models/v1_team.py +1 -1
- polyaxon_sdk/models/v1_team_member.py +1 -1
- polyaxon_sdk/models/v1_team_settings.py +1 -1
- polyaxon_sdk/models/v1_template.py +1 -1
- polyaxon_sdk/models/v1_tensorboard_type.py +1 -1
- polyaxon_sdk/models/v1_termination.py +1 -1
- polyaxon_sdk/models/v1_tf_job.py +4 -2
- polyaxon_sdk/models/v1_token.py +1 -1
- polyaxon_sdk/models/v1_trial_start.py +1 -1
- polyaxon_sdk/models/v1_trigger_policy.py +1 -1
- polyaxon_sdk/models/v1_truncation_stopping_policy.py +1 -1
- polyaxon_sdk/models/v1_tuner.py +1 -1
- polyaxon_sdk/models/v1_uri_type.py +1 -1
- polyaxon_sdk/models/v1_url_ref.py +1 -1
- polyaxon_sdk/models/v1_user.py +1 -1
- polyaxon_sdk/models/v1_user_email.py +1 -1
- polyaxon_sdk/models/v1_user_singup.py +1 -1
- polyaxon_sdk/models/v1_uuids.py +1 -1
- polyaxon_sdk/models/v1_validation.py +1 -1
- polyaxon_sdk/models/v1_version.py +1 -1
- polyaxon_sdk/models/v1_wasb_type.py +1 -1
- polyaxon_sdk/models/v1_xg_boost_job.py +1 -1
- polyaxon_sdk/rest.py +1 -1
- {polyaxon_sdk-2.1.0rc3.dist-info → polyaxon_sdk-2.1.0rc5.dist-info}/METADATA +1 -1
- polyaxon_sdk-2.1.0rc5.dist-info/RECORD +260 -0
- polyaxon_sdk-2.1.0rc3.dist-info/RECORD +0 -258
- {polyaxon_sdk-2.1.0rc3.dist-info → polyaxon_sdk-2.1.0rc5.dist-info}/WHEEL +0 -0
- {polyaxon_sdk-2.1.0rc3.dist-info → polyaxon_sdk-2.1.0rc5.dist-info}/top_level.txt +0 -0
|
@@ -5,7 +5,7 @@
|
|
|
5
5
|
|
|
6
6
|
# noqa: E501
|
|
7
7
|
|
|
8
|
-
The version of the OpenAPI document: 2.1.0-
|
|
8
|
+
The version of the OpenAPI document: 2.1.0-rc5
|
|
9
9
|
Contact: contact@polyaxon.com
|
|
10
10
|
Generated by OpenAPI Generator (https://openapi-generator.tech)
|
|
11
11
|
|
|
@@ -24,6 +24,7 @@ from typing import Optional
|
|
|
24
24
|
from pydantic import BaseModel, Field, StrictStr
|
|
25
25
|
from polyaxon_sdk.models.v1_clean_pod_policy import V1CleanPodPolicy
|
|
26
26
|
from polyaxon_sdk.models.v1_kf_replica import V1KFReplica
|
|
27
|
+
from polyaxon_sdk.models.v1_paddle_elastic_polic import V1PaddleElasticPolic
|
|
27
28
|
from polyaxon_sdk.models.v1_scheduling_policy import V1SchedulingPolicy
|
|
28
29
|
|
|
29
30
|
class V1PaddleJob(BaseModel):
|
|
@@ -33,9 +34,10 @@ class V1PaddleJob(BaseModel):
|
|
|
33
34
|
kind: Optional[StrictStr] = 'paddlejob'
|
|
34
35
|
clean_pod_policy: Optional[V1CleanPodPolicy] = Field(None, alias="cleanPodPolicy")
|
|
35
36
|
scheduling_policy: Optional[V1SchedulingPolicy] = Field(None, alias="schedulingPolicy")
|
|
37
|
+
elastic_policy: Optional[V1PaddleElasticPolic] = Field(None, alias="elasticPolicy")
|
|
36
38
|
master: Optional[V1KFReplica] = None
|
|
37
39
|
worker: Optional[V1KFReplica] = None
|
|
38
|
-
__properties = ["kind", "cleanPodPolicy", "schedulingPolicy", "master", "worker"]
|
|
40
|
+
__properties = ["kind", "cleanPodPolicy", "schedulingPolicy", "elasticPolicy", "master", "worker"]
|
|
39
41
|
|
|
40
42
|
class Config:
|
|
41
43
|
allow_population_by_field_name = True
|
|
@@ -63,6 +65,9 @@ class V1PaddleJob(BaseModel):
|
|
|
63
65
|
# override the default output from pydantic by calling `to_dict()` of scheduling_policy
|
|
64
66
|
if self.scheduling_policy:
|
|
65
67
|
_dict['schedulingPolicy'] = self.scheduling_policy.to_dict()
|
|
68
|
+
# override the default output from pydantic by calling `to_dict()` of elastic_policy
|
|
69
|
+
if self.elastic_policy:
|
|
70
|
+
_dict['elasticPolicy'] = self.elastic_policy.to_dict()
|
|
66
71
|
# override the default output from pydantic by calling `to_dict()` of master
|
|
67
72
|
if self.master:
|
|
68
73
|
_dict['master'] = self.master.to_dict()
|
|
@@ -84,6 +89,7 @@ class V1PaddleJob(BaseModel):
|
|
|
84
89
|
"kind": obj.get("kind") if obj.get("kind") is not None else 'paddlejob',
|
|
85
90
|
"clean_pod_policy": obj.get("cleanPodPolicy"),
|
|
86
91
|
"scheduling_policy": V1SchedulingPolicy.from_dict(obj.get("schedulingPolicy")) if obj.get("schedulingPolicy") is not None else None,
|
|
92
|
+
"elastic_policy": V1PaddleElasticPolic.from_dict(obj.get("elasticPolicy")) if obj.get("elasticPolicy") is not None else None,
|
|
87
93
|
"master": V1KFReplica.from_dict(obj.get("master")) if obj.get("master") is not None else None,
|
|
88
94
|
"worker": V1KFReplica.from_dict(obj.get("worker")) if obj.get("worker") is not None else None
|
|
89
95
|
})
|
polyaxon_sdk/models/v1_param.py
CHANGED
polyaxon_sdk/models/v1_preset.py
CHANGED
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
# coding: utf-8
|
|
2
|
+
|
|
3
|
+
"""
|
|
4
|
+
Polyaxon SDKs and REST API specification.
|
|
5
|
+
|
|
6
|
+
# noqa: E501
|
|
7
|
+
|
|
8
|
+
The version of the OpenAPI document: 2.1.0-rc5
|
|
9
|
+
Contact: contact@polyaxon.com
|
|
10
|
+
Generated by OpenAPI Generator (https://openapi-generator.tech)
|
|
11
|
+
|
|
12
|
+
Do not edit the class manually.
|
|
13
|
+
"""
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
from __future__ import annotations
|
|
17
|
+
from inspect import getfullargspec
|
|
18
|
+
import pprint
|
|
19
|
+
import re # noqa: F401
|
|
20
|
+
import json
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
from typing import Any, Dict, List, Optional
|
|
24
|
+
from pydantic import BaseModel, Field, StrictBool, StrictInt, StrictStr, conlist
|
|
25
|
+
|
|
26
|
+
class V1PytorchElasticPolicy(BaseModel):
|
|
27
|
+
"""
|
|
28
|
+
V1PytorchElasticPolicy
|
|
29
|
+
"""
|
|
30
|
+
min_replicas: Optional[StrictInt] = Field(None, alias="minReplicas", description="minReplicas is the lower limit for the number of replicas to which the training job can scale down.")
|
|
31
|
+
max_replicas: Optional[StrictInt] = Field(None, alias="maxReplicas", description="upper limit for the number of pods that can be set by the autoscaler; cannot be smaller than MinReplicas, defaults to null.")
|
|
32
|
+
rdvz_backend: Optional[StrictStr] = Field(None, alias="rdvzBackend")
|
|
33
|
+
rdvz_port: Optional[StrictInt] = Field(None, alias="rdvzPort")
|
|
34
|
+
rdvz_host: Optional[StrictStr] = Field(None, alias="rdvzHost")
|
|
35
|
+
rdvz_id: Optional[StrictStr] = Field(None, alias="rdvzId")
|
|
36
|
+
rdvz_conf: Optional[conlist(Dict[str, Any])] = Field(None, alias="rdvzConf", description="RDZVConf contains additional rendezvous configuration (<key1>=<value1>,<key2>=<value2>,...).")
|
|
37
|
+
standalone: Optional[StrictBool] = Field(None, description="Start a local standalone rendezvous backend that is represented by a C10d TCP store on port 29400. Useful when launching single-node, multi-worker job. If specified --rdzv_backend, --rdzv_endpoint, --rdzv_id are auto-assigned; any explicitly set values are ignored.")
|
|
38
|
+
n_proc_per_node: Optional[StrictInt] = Field(None, alias="nProcPerNode", description="Number of workers per node; supported values: [auto, cpu, gpu, int]. Deprecated: This API is deprecated in v1.7+ Use .spec.nprocPerNode instead.")
|
|
39
|
+
max_restarts: Optional[StrictInt] = Field(None, alias="maxRestarts")
|
|
40
|
+
metrics: Optional[conlist(Dict[str, Any])] = Field(None, alias="Metrics", description="Metrics contains the specifications which are used to calculate the desired replica count (the maximum replica count across all metrics will be used). The desired replica count is calculated with multiplying the ratio between the target value and the current value by the current number of pods. Ergo, metrics used must decrease as the pod count is increased, and vice-versa. See the individual metric source types for more information about how each type of metric must respond. If not set, the HPA will not be created.")
|
|
41
|
+
__properties = ["minReplicas", "maxReplicas", "rdvzBackend", "rdvzPort", "rdvzHost", "rdvzId", "rdvzConf", "standalone", "nProcPerNode", "maxRestarts", "Metrics"]
|
|
42
|
+
|
|
43
|
+
class Config:
|
|
44
|
+
allow_population_by_field_name = True
|
|
45
|
+
validate_assignment = True
|
|
46
|
+
|
|
47
|
+
def to_str(self) -> str:
|
|
48
|
+
"""Returns the string representation of the model using alias"""
|
|
49
|
+
return pprint.pformat(self.dict(by_alias=True))
|
|
50
|
+
|
|
51
|
+
def to_json(self) -> str:
|
|
52
|
+
"""Returns the JSON representation of the model using alias"""
|
|
53
|
+
return json.dumps(self.to_dict())
|
|
54
|
+
|
|
55
|
+
@classmethod
|
|
56
|
+
def from_json(cls, json_str: str) -> V1PytorchElasticPolicy:
|
|
57
|
+
"""Create an instance of V1PytorchElasticPolicy from a JSON string"""
|
|
58
|
+
return cls.from_dict(json.loads(json_str))
|
|
59
|
+
|
|
60
|
+
def to_dict(self):
|
|
61
|
+
"""Returns the dictionary representation of the model using alias"""
|
|
62
|
+
_dict = self.dict(by_alias=True,
|
|
63
|
+
exclude={
|
|
64
|
+
},
|
|
65
|
+
exclude_none=True)
|
|
66
|
+
return _dict
|
|
67
|
+
|
|
68
|
+
@classmethod
|
|
69
|
+
def from_dict(cls, obj: dict) -> V1PytorchElasticPolicy:
|
|
70
|
+
"""Create an instance of V1PytorchElasticPolicy from a dict"""
|
|
71
|
+
if obj is None:
|
|
72
|
+
return None
|
|
73
|
+
|
|
74
|
+
if type(obj) is not dict:
|
|
75
|
+
return V1PytorchElasticPolicy.parse_obj(obj)
|
|
76
|
+
|
|
77
|
+
_obj = V1PytorchElasticPolicy.parse_obj({
|
|
78
|
+
"min_replicas": obj.get("minReplicas"),
|
|
79
|
+
"max_replicas": obj.get("maxReplicas"),
|
|
80
|
+
"rdvz_backend": obj.get("rdvzBackend"),
|
|
81
|
+
"rdvz_port": obj.get("rdvzPort"),
|
|
82
|
+
"rdvz_host": obj.get("rdvzHost"),
|
|
83
|
+
"rdvz_id": obj.get("rdvzId"),
|
|
84
|
+
"rdvz_conf": obj.get("rdvzConf"),
|
|
85
|
+
"standalone": obj.get("standalone"),
|
|
86
|
+
"n_proc_per_node": obj.get("nProcPerNode"),
|
|
87
|
+
"max_restarts": obj.get("maxRestarts"),
|
|
88
|
+
"metrics": obj.get("Metrics")
|
|
89
|
+
})
|
|
90
|
+
return _obj
|
|
91
|
+
|
|
@@ -5,7 +5,7 @@
|
|
|
5
5
|
|
|
6
6
|
# noqa: E501
|
|
7
7
|
|
|
8
|
-
The version of the OpenAPI document: 2.1.0-
|
|
8
|
+
The version of the OpenAPI document: 2.1.0-rc5
|
|
9
9
|
Contact: contact@polyaxon.com
|
|
10
10
|
Generated by OpenAPI Generator (https://openapi-generator.tech)
|
|
11
11
|
|
|
@@ -24,6 +24,7 @@ from typing import Optional
|
|
|
24
24
|
from pydantic import BaseModel, Field, StrictStr
|
|
25
25
|
from polyaxon_sdk.models.v1_clean_pod_policy import V1CleanPodPolicy
|
|
26
26
|
from polyaxon_sdk.models.v1_kf_replica import V1KFReplica
|
|
27
|
+
from polyaxon_sdk.models.v1_pytorch_elastic_policy import V1PytorchElasticPolicy
|
|
27
28
|
from polyaxon_sdk.models.v1_scheduling_policy import V1SchedulingPolicy
|
|
28
29
|
|
|
29
30
|
class V1PytorchJob(BaseModel):
|
|
@@ -33,9 +34,11 @@ class V1PytorchJob(BaseModel):
|
|
|
33
34
|
kind: Optional[StrictStr] = 'pytorchjob'
|
|
34
35
|
clean_pod_policy: Optional[V1CleanPodPolicy] = Field(None, alias="cleanPodPolicy")
|
|
35
36
|
scheduling_policy: Optional[V1SchedulingPolicy] = Field(None, alias="schedulingPolicy")
|
|
37
|
+
elastic_policy: Optional[V1PytorchElasticPolicy] = Field(None, alias="elasticPolicy")
|
|
36
38
|
master: Optional[V1KFReplica] = None
|
|
37
39
|
worker: Optional[V1KFReplica] = None
|
|
38
|
-
|
|
40
|
+
n_proc_per_node: Optional[StrictStr] = Field(None, alias="nProcPerNode")
|
|
41
|
+
__properties = ["kind", "cleanPodPolicy", "schedulingPolicy", "elasticPolicy", "master", "worker", "nProcPerNode"]
|
|
39
42
|
|
|
40
43
|
class Config:
|
|
41
44
|
allow_population_by_field_name = True
|
|
@@ -63,6 +66,9 @@ class V1PytorchJob(BaseModel):
|
|
|
63
66
|
# override the default output from pydantic by calling `to_dict()` of scheduling_policy
|
|
64
67
|
if self.scheduling_policy:
|
|
65
68
|
_dict['schedulingPolicy'] = self.scheduling_policy.to_dict()
|
|
69
|
+
# override the default output from pydantic by calling `to_dict()` of elastic_policy
|
|
70
|
+
if self.elastic_policy:
|
|
71
|
+
_dict['elasticPolicy'] = self.elastic_policy.to_dict()
|
|
66
72
|
# override the default output from pydantic by calling `to_dict()` of master
|
|
67
73
|
if self.master:
|
|
68
74
|
_dict['master'] = self.master.to_dict()
|
|
@@ -84,8 +90,10 @@ class V1PytorchJob(BaseModel):
|
|
|
84
90
|
"kind": obj.get("kind") if obj.get("kind") is not None else 'pytorchjob',
|
|
85
91
|
"clean_pod_policy": obj.get("cleanPodPolicy"),
|
|
86
92
|
"scheduling_policy": V1SchedulingPolicy.from_dict(obj.get("schedulingPolicy")) if obj.get("schedulingPolicy") is not None else None,
|
|
93
|
+
"elastic_policy": V1PytorchElasticPolicy.from_dict(obj.get("elasticPolicy")) if obj.get("elasticPolicy") is not None else None,
|
|
87
94
|
"master": V1KFReplica.from_dict(obj.get("master")) if obj.get("master") is not None else None,
|
|
88
|
-
"worker": V1KFReplica.from_dict(obj.get("worker")) if obj.get("worker") is not None else None
|
|
95
|
+
"worker": V1KFReplica.from_dict(obj.get("worker")) if obj.get("worker") is not None else None,
|
|
96
|
+
"n_proc_per_node": obj.get("nProcPerNode")
|
|
89
97
|
})
|
|
90
98
|
return _obj
|
|
91
99
|
|
polyaxon_sdk/models/v1_queue.py
CHANGED
polyaxon_sdk/models/v1_run.py
CHANGED
|
@@ -5,7 +5,7 @@
|
|
|
5
5
|
|
|
6
6
|
# noqa: E501
|
|
7
7
|
|
|
8
|
-
The version of the OpenAPI document: 2.1.0-
|
|
8
|
+
The version of the OpenAPI document: 2.1.0-rc5
|
|
9
9
|
Contact: contact@polyaxon.com
|
|
10
10
|
Generated by OpenAPI Generator (https://openapi-generator.tech)
|
|
11
11
|
|
|
@@ -20,7 +20,7 @@ import re # noqa: F401
|
|
|
20
20
|
import json
|
|
21
21
|
|
|
22
22
|
|
|
23
|
-
from typing import Optional
|
|
23
|
+
from typing import Any, Dict, Optional
|
|
24
24
|
from pydantic import BaseModel, Field, StrictInt, StrictStr
|
|
25
25
|
|
|
26
26
|
class V1SchedulingPolicy(BaseModel):
|
|
@@ -29,8 +29,10 @@ class V1SchedulingPolicy(BaseModel):
|
|
|
29
29
|
"""
|
|
30
30
|
min_available: Optional[StrictInt] = Field(None, alias="minAvailable")
|
|
31
31
|
queue: Optional[StrictStr] = None
|
|
32
|
+
min_resources: Optional[Dict[str, Any]] = Field(None, alias="minResources")
|
|
32
33
|
priority_class: Optional[StrictStr] = Field(None, alias="priorityClass")
|
|
33
|
-
|
|
34
|
+
schedule_timeout_seconds: Optional[StrictInt] = Field(None, alias="scheduleTimeoutSeconds")
|
|
35
|
+
__properties = ["minAvailable", "queue", "minResources", "priorityClass", "scheduleTimeoutSeconds"]
|
|
34
36
|
|
|
35
37
|
class Config:
|
|
36
38
|
allow_population_by_field_name = True
|
|
@@ -69,7 +71,9 @@ class V1SchedulingPolicy(BaseModel):
|
|
|
69
71
|
_obj = V1SchedulingPolicy.parse_obj({
|
|
70
72
|
"min_available": obj.get("minAvailable"),
|
|
71
73
|
"queue": obj.get("queue"),
|
|
72
|
-
"
|
|
74
|
+
"min_resources": obj.get("minResources"),
|
|
75
|
+
"priority_class": obj.get("priorityClass"),
|
|
76
|
+
"schedule_timeout_seconds": obj.get("scheduleTimeoutSeconds")
|
|
73
77
|
})
|
|
74
78
|
return _obj
|
|
75
79
|
|
polyaxon_sdk/models/v1_search.py
CHANGED