|
@@ -1,4 +1,5 @@
|
|
|
-using System.Windows;
|
|
|
+using System;
|
|
|
+using System.Windows;
|
|
|
using AIPlatform.Protocol.Entities;
|
|
|
|
|
|
namespace aipdev
|
|
@@ -22,9 +23,10 @@ namespace aipdev
|
|
|
{
|
|
|
InitializeComponent();
|
|
|
PackageName.Focus();
|
|
|
- PackageFramework.Items.Add(TrainScriptPackageFramework.Tensorflow);
|
|
|
- PackageFramework.Items.Add(TrainScriptPackageFramework.PyTorch);
|
|
|
- PackageFramework.Items.Add(TrainScriptPackageFramework.CNTK);
|
|
|
+ foreach (TrainScriptPackageFramework type in Enum.GetValues(typeof(TrainScriptPackageFramework)))
|
|
|
+ {
|
|
|
+ PackageFramework.Items.Add(type);
|
|
|
+ }
|
|
|
PackageFramework.SelectedIndex = 1;
|
|
|
}
|
|
|
|
|
@@ -32,11 +34,11 @@ namespace aipdev
|
|
|
{
|
|
|
if (string.IsNullOrWhiteSpace(PackageName.Text) || string.IsNullOrEmpty(PackageName.Text))
|
|
|
{
|
|
|
- MessageBox.Show(Application.Current.MainWindow, "请输入脚本包名","错误",MessageBoxButton.OK, MessageBoxImage.Error);
|
|
|
+ MessageBox.Show(Application.Current.MainWindow, "请输入脚本包名", "错误", MessageBoxButton.OK, MessageBoxImage.Error);
|
|
|
return;
|
|
|
}
|
|
|
NewPackageName = PackageName.Text.Trim();
|
|
|
- NewPackageFramework = (TrainScriptPackageFramework) PackageFramework.SelectedItem;
|
|
|
+ NewPackageFramework = (TrainScriptPackageFramework)PackageFramework.SelectedItem;
|
|
|
Close();
|
|
|
}
|
|
|
|