Skip to content

Commit 09afe06

Browse files
author
fengyikai
committed
星流平台(aicp):训练任务openapi相关字段校正
1 parent 1299a74 commit 09afe06

2 files changed

Lines changed: 17 additions & 9 deletions

File tree

ksyun/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,4 @@
1414
# limitations under the License.
1515

1616

17-
__version__ = '1.5.8.24'
17+
__version__ = '1.5.8.25'

ksyun/client/aicp/v20240612/models.py

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -551,6 +551,11 @@ def __init__(self):
551551
:type PathPrefix: String
552552
:param FrameworkReplicas: 框架副本配置
553553
:type PathPrefix: Object
554+
:param RestartPolicy: 重启策略,可选值:
555+
· Always
556+
· OnFailure
557+
· Never
558+
:type PathPrefix: String
554559
:param Envs: 环境变量
555560
:type PathPrefix: Array
556561
:param SupportTensorboard: 是否开启Tensorboard,默认不开启
@@ -567,7 +572,7 @@ def __init__(self):
567572
:type PathPrefix: String
568573
:param GPUNumber: 范围0-10000
569574
:type PathPrefix: Int
570-
:param CpuNum: 范围0-10000
575+
:param CPUNum: 范围0-10000
571576
:type PathPrefix: Int
572577
:param Memory: 0-10000
573578
:type PathPrefix: Int
@@ -580,7 +585,7 @@ def __init__(self):
580585
:type PathPrefix: Int
581586
:param SelfHealing: 是否开启自愈,默认true
582587
:type PathPrefix: Boolean
583-
:param RunOnCpu: 任务是否仅运行在cpu节点上,默认false
588+
:param RunOnCPU: 任务是否仅运行在cpu节点上,默认false
584589
:type PathPrefix: Boolean
585590
"""
586591
self.TrainJobName = None
@@ -592,6 +597,7 @@ def __init__(self):
592597
self.Framework = None
593598
self.ImageSource = None
594599
self.FrameworkReplicas = None
600+
self.RestartPolicy = None
595601
self.Envs = None
596602
self.SupportTensorboard = None
597603
self.ImageId = None
@@ -600,13 +606,13 @@ def __init__(self):
600606
self.ImageTagId = None
601607
self.GPUType = None
602608
self.GPUNumber = None
603-
self.CpuNum = None
609+
self.CPUNum = None
604610
self.Memory = None
605611
self.StorageConfigs = None
606612
self.AccessType = None
607613
self.MaxRuntime = None
608614
self.SelfHealing = None
609-
self.RunOnCpu = None
615+
self.RunOnCPU = None
610616

611617
def _deserialize(self, params):
612618
if params.get("TrainJobName"):
@@ -627,6 +633,8 @@ def _deserialize(self, params):
627633
self.ImageSource = params.get("ImageSource")
628634
if params.get("FrameworkReplicas"):
629635
self.FrameworkReplicas = params.get("FrameworkReplicas")
636+
if params.get("RestartPolicy"):
637+
self.RestartPolicy = params.get("RestartPolicy")
630638
if params.get("Envs"):
631639
self.Envs = params.get("Envs")
632640
if params.get("SupportTensorboard"):
@@ -643,8 +651,8 @@ def _deserialize(self, params):
643651
self.GPUType = params.get("GPUType")
644652
if params.get("GPUNumber"):
645653
self.GPUNumber = params.get("GPUNumber")
646-
if params.get("CpuNum"):
647-
self.CpuNum = params.get("CpuNum")
654+
if params.get("CPUNum"):
655+
self.CPUNum = params.get("CPUNum")
648656
if params.get("Memory"):
649657
self.Memory = params.get("Memory")
650658
if params.get("StorageConfigs"):
@@ -655,8 +663,8 @@ def _deserialize(self, params):
655663
self.MaxRuntime = params.get("MaxRuntime")
656664
if params.get("SelfHealing"):
657665
self.SelfHealing = params.get("SelfHealing")
658-
if params.get("RunOnCpu"):
659-
self.RunOnCpu = params.get("RunOnCpu")
666+
if params.get("RunOnCPU"):
667+
self.RunOnCPU = params.get("RunOnCPU")
660668

661669

662670
class DescribeTrainJobEventsRequest(AbstractModel):

0 commit comments

Comments
 (0)