Skip to content

Commit

Permalink
generate codes
Browse files Browse the repository at this point in the history 
Signed-off-by: Yuki Iwai <[email protected]>
  • Loading branch information
@tenzen-y
tenzen-y committed Jan 15, 2023
1 parent f9fa8ac commit 2ae0b6a
Show file tree
Hide file tree
Showing 30 changed files with 120 additions and 34 deletions.
4 changes: 4 additions & 0 deletions hack/python-sdk/swagger.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -773,6 +773,10 @@
},
"queue": {
"type": "string"
},
"scheduleTimeoutSeconds": {
"type": "integer",
"format": "int32"
}
}
}
Expand Down
3 changes: 3 additions & 0 deletions manifests/base/crds/kubeflow.org_mpijobs.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7870,6 +7870,9 @@ spec:
type: string
queue:
type: string
scheduleTimeoutSeconds:
format: int32
type: integer
type: object
ttlSecondsAfterFinished:
description: TTLSecondsAfterFinished is the TTL to clean up jobs.
Expand Down
3 changes: 3 additions & 0 deletions manifests/base/crds/kubeflow.org_mxjobs.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7870,6 +7870,9 @@ spec:
type: string
queue:
type: string
scheduleTimeoutSeconds:
format: int32
type: integer
type: object
ttlSecondsAfterFinished:
description: TTLSecondsAfterFinished is the TTL to clean up jobs.
Expand Down
3 changes: 3 additions & 0 deletions manifests/base/crds/kubeflow.org_paddlejobs.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8380,6 +8380,9 @@ spec:
type: string
queue:
type: string
scheduleTimeoutSeconds:
format: int32
type: integer
type: object
ttlSecondsAfterFinished:
description: TTLSecondsAfterFinished is the TTL to clean up jobs.
Expand Down
3 changes: 3 additions & 0 deletions manifests/base/crds/kubeflow.org_pytorchjobs.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8409,6 +8409,9 @@ spec:
type: string
queue:
type: string
scheduleTimeoutSeconds:
format: int32
type: integer
type: object
ttlSecondsAfterFinished:
description: TTLSecondsAfterFinished is the TTL to clean up jobs.
Expand Down
3 changes: 3 additions & 0 deletions manifests/base/crds/kubeflow.org_tfjobs.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,9 @@ spec:
type: string
queue:
type: string
scheduleTimeoutSeconds:
format: int32
type: integer
type: object
ttlSecondsAfterFinished:
description: TTLSecondsAfterFinished is the TTL to clean up jobs.
Expand Down
3 changes: 3 additions & 0 deletions manifests/base/crds/kubeflow.org_xgboostjobs.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,9 @@ spec:
type: string
queue:
type: string
scheduleTimeoutSeconds:
format: int32
type: integer
type: object
ttlSecondsAfterFinished:
description: TTLSecondsAfterFinished is the TTL to clean up jobs.
Expand Down
6 changes: 6 additions & 0 deletions pkg/apis/kubeflow.org/v1/openapi_generated.go
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions sdk/python/docs/V1SchedulingPolicy.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ Name | Type | Description | Notes
**min_resources** | [**dict(str, Quantity)**](Quantity.md) | | [optional]
**priority_class** | **str** | | [optional]
**queue** | **str** | | [optional]
**schedule_timeout_seconds** | **int** | | [optional]

[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)

Expand Down
32 changes: 29 additions & 3 deletions sdk/python/kubeflow/training/models/v1_scheduling_policy.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -36,17 +36,19 @@ class V1SchedulingPolicy(object):
'min_available': 'int',
'min_resources': 'dict(str, Quantity)',
'priority_class': 'str',
'queue': 'str'
'queue': 'str',
'schedule_timeout_seconds': 'int'
}

attribute_map = {
'min_available': 'minAvailable',
'min_resources': 'minResources',
'priority_class': 'priorityClass',
'queue': 'queue'
'queue': 'queue',
'schedule_timeout_seconds': 'scheduleTimeoutSeconds'
}

def __init__(self, min_available=None, min_resources=None, priority_class=None, queue=None, local_vars_configuration=None): # noqa: E501
def __init__(self, min_available=None, min_resources=None, priority_class=None, queue=None, schedule_timeout_seconds=None, local_vars_configuration=None): # noqa: E501
"""V1SchedulingPolicy - a model defined in OpenAPI""" # noqa: E501
if local_vars_configuration is None:
local_vars_configuration = Configuration()
Expand All @@ -56,6 +58,7 @@ def __init__(self, min_available=None, min_resources=None, priority_class=None,
self._min_resources = None
self._priority_class = None
self._queue = None
self._schedule_timeout_seconds = None
self.discriminator = None

if min_available is not None:
Expand All @@ -66,6 +69,8 @@ def __init__(self, min_available=None, min_resources=None, priority_class=None,
self.priority_class = priority_class
if queue is not None:
self.queue = queue
if schedule_timeout_seconds is not None:
self.schedule_timeout_seconds = schedule_timeout_seconds

@property
def min_available(self):
Expand Down Expand Up @@ -151,6 +156,27 @@ def queue(self, queue):

self._queue = queue

@property
def schedule_timeout_seconds(self):
"""Gets the schedule_timeout_seconds of this V1SchedulingPolicy. # noqa: E501


:return: The schedule_timeout_seconds of this V1SchedulingPolicy. # noqa: E501
:rtype: int
"""
return self._schedule_timeout_seconds

@schedule_timeout_seconds.setter
def schedule_timeout_seconds(self, schedule_timeout_seconds):
"""Sets the schedule_timeout_seconds of this V1SchedulingPolicy.


:param schedule_timeout_seconds: The schedule_timeout_seconds of this V1SchedulingPolicy. # noqa: E501
:type: int
"""

self._schedule_timeout_seconds = schedule_timeout_seconds

def to_dict(self):
"""Returns the model properties as a dict"""
result = {}
Expand Down
3 changes: 2 additions & 1 deletion sdk/python/test/test_kubeflow_org_v1_mpi_job.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
slots_per_worker = 56, ),
status = V1JobStatus(
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_mpi_job_list.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
slots_per_worker = 56, ),
status = V1JobStatus(
Expand Down Expand Up @@ -114,7 +115,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
slots_per_worker = 56, ),
status = V1JobStatus(
Expand Down
3 changes: 2 additions & 1 deletion sdk/python/test/test_kubeflow_org_v1_mpi_job_spec.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
slots_per_worker = 56
)
Expand Down
3 changes: 2 additions & 1 deletion sdk/python/test/test_kubeflow_org_v1_mx_job.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_mx_job_list.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down Expand Up @@ -111,7 +112,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_mx_job_spec.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, )
)
else :
Expand All @@ -75,7 +76,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
)

Expand Down
3 changes: 2 additions & 1 deletion sdk/python/test/test_kubeflow_org_v1_paddle_job.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_paddle_job_list.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down Expand Up @@ -123,7 +124,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_paddle_job_spec.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, )
)
else :
Expand All @@ -80,7 +81,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
)

Expand Down
3 changes: 2 additions & 1 deletion sdk/python/test/test_kubeflow_org_v1_py_torch_job.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_py_torch_job_list.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down Expand Up @@ -145,7 +146,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ), ),
status = V1JobStatus(
completion_time = None,
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_py_torch_job_spec.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, )
)
else :
Expand All @@ -91,7 +92,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
)

Expand Down
3 changes: 2 additions & 1 deletion sdk/python/test/test_kubeflow_org_v1_tf_job.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
success_policy = '0',
tf_replica_specs = {
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_tf_job_list.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
success_policy = '0',
tf_replica_specs = {
Expand Down Expand Up @@ -106,7 +107,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
success_policy = '0',
tf_replica_specs = {
Expand Down
6 changes: 4 additions & 2 deletions sdk/python/test/test_kubeflow_org_v1_tf_job_spec.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
success_policy = '0',
tf_replica_specs = {
Expand All @@ -69,7 +70,8 @@ def make_instance(self, include_optional):
'key' : None
},
priority_class = '0',
queue = '0', ),
queue = '0',
schedule_timeout_seconds = 56, ),
ttl_seconds_after_finished = 56, ),
tf_replica_specs = {
'key' : V1ReplicaSpec(
Expand Down
Loading

0 comments on commit 2ae0b6a

Please sign in to comment.