Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
72d04abb
Unverified
Commit
72d04abb
authored
Dec 10, 2019
by
chicm-ms
Committed by
GitHub
Dec 10, 2019
Browse files
refactor code to fix eslint error (#1835)
parent
ef9c8b15
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
58 additions
and
58 deletions
+58
-58
src/nni_manager/training_service/kubernetes/frameworkcontroller/frameworkcontrollerApiClient.ts
...netes/frameworkcontroller/frameworkcontrollerApiClient.ts
+4
-4
src/nni_manager/training_service/kubernetes/frameworkcontroller/frameworkcontrollerTrainingService.ts
...frameworkcontroller/frameworkcontrollerTrainingService.ts
+2
-2
src/nni_manager/training_service/kubernetes/kubeflow/kubeflowApiClient.ts
...training_service/kubernetes/kubeflow/kubeflowApiClient.ts
+49
-49
src/nni_manager/training_service/kubernetes/kubeflow/kubeflowTrainingService.ts
...ng_service/kubernetes/kubeflow/kubeflowTrainingService.ts
+3
-3
No files found.
src/nni_manager/training_service/kubernetes/frameworkcontroller/frameworkcontrollerApiClient.ts
View file @
72d04abb
...
@@ -9,12 +9,12 @@ import { GeneralK8sClient, KubernetesCRDClient } from '../kubernetesApiClient';
...
@@ -9,12 +9,12 @@ import { GeneralK8sClient, KubernetesCRDClient } from '../kubernetesApiClient';
/**
/**
* FrameworkController Client
* FrameworkController Client
*/
*/
abstract
class
FrameworkControllerClient
extends
KubernetesCRDClient
{
class
FrameworkControllerClient
Factory
{
/**
/**
* Factory method to generate operator client
* Factory method to generate operator client
*/
*/
// tslint:disable-next-line:function-name
// tslint:disable-next-line:function-name
public
static
generateFrameworkController
Client
():
KubernetesCRDClient
{
public
static
create
Client
():
KubernetesCRDClient
{
return
new
FrameworkControllerClientV1
();
return
new
FrameworkControllerClientV1
();
}
}
}
}
...
@@ -22,7 +22,7 @@ abstract class FrameworkControllerClient extends KubernetesCRDClient {
...
@@ -22,7 +22,7 @@ abstract class FrameworkControllerClient extends KubernetesCRDClient {
/**
/**
* FrameworkController ClientV1
* FrameworkController ClientV1
*/
*/
class
FrameworkControllerClientV1
extends
FrameworkController
Client
{
class
FrameworkControllerClientV1
extends
KubernetesCRD
Client
{
/**
/**
* constructor, to initialize frameworkcontroller CRD definition
* constructor, to initialize frameworkcontroller CRD definition
*/
*/
...
@@ -43,4 +43,4 @@ class FrameworkControllerClientV1 extends FrameworkControllerClient {
...
@@ -43,4 +43,4 @@ class FrameworkControllerClientV1 extends FrameworkControllerClient {
}
}
}
}
export
{
FrameworkControllerClient
,
GeneralK8sClient
};
export
{
FrameworkControllerClient
Factory
,
GeneralK8sClient
};
src/nni_manager/training_service/kubernetes/frameworkcontroller/frameworkcontrollerTrainingService.ts
View file @
72d04abb
...
@@ -19,7 +19,7 @@ import { AzureStorageClientUtility } from '../azureStorageClientUtils';
...
@@ -19,7 +19,7 @@ import { AzureStorageClientUtility } from '../azureStorageClientUtils';
import
{
NFSConfig
}
from
'
../kubernetesConfig
'
;
import
{
NFSConfig
}
from
'
../kubernetesConfig
'
;
import
{
KubernetesTrialJobDetail
}
from
'
../kubernetesData
'
;
import
{
KubernetesTrialJobDetail
}
from
'
../kubernetesData
'
;
import
{
KubernetesTrainingService
}
from
'
../kubernetesTrainingService
'
;
import
{
KubernetesTrainingService
}
from
'
../kubernetesTrainingService
'
;
import
{
FrameworkControllerClient
}
from
'
./frameworkcontrollerApiClient
'
;
import
{
FrameworkControllerClient
Factory
}
from
'
./frameworkcontrollerApiClient
'
;
import
{
FrameworkControllerClusterConfig
,
FrameworkControllerClusterConfigAzure
,
FrameworkControllerClusterConfigFactory
,
import
{
FrameworkControllerClusterConfig
,
FrameworkControllerClusterConfigAzure
,
FrameworkControllerClusterConfigFactory
,
FrameworkControllerClusterConfigNFS
,
FrameworkControllerTrialConfig
}
from
'
./frameworkcontrollerConfig
'
;
FrameworkControllerClusterConfigNFS
,
FrameworkControllerTrialConfig
}
from
'
./frameworkcontrollerConfig
'
;
import
{
FrameworkControllerJobInfoCollector
}
from
'
./frameworkcontrollerJobInfoCollector
'
;
import
{
FrameworkControllerJobInfoCollector
}
from
'
./frameworkcontrollerJobInfoCollector
'
;
...
@@ -142,7 +142,7 @@ class FrameworkControllerTrainingService extends KubernetesTrainingService imple
...
@@ -142,7 +142,7 @@ class FrameworkControllerTrainingService extends KubernetesTrainingService imple
nfsFrameworkControllerClusterConfig
.
nfs
.
path
nfsFrameworkControllerClusterConfig
.
nfs
.
path
);
);
}
}
this
.
kubernetesCRDClient
=
FrameworkControllerClient
.
generateFrameworkController
Client
();
this
.
kubernetesCRDClient
=
FrameworkControllerClient
Factory
.
create
Client
();
break
;
break
;
case
TrialConfigMetadataKey
.
TRIAL_CONFIG
:
case
TrialConfigMetadataKey
.
TRIAL_CONFIG
:
const
frameworkcontrollerTrialJsonObjsect
:
any
=
JSON
.
parse
(
value
);
const
frameworkcontrollerTrialJsonObjsect
:
any
=
JSON
.
parse
(
value
);
...
...
src/nni_manager/training_service/kubernetes/kubeflow/kubeflowApiClient.ts
View file @
72d04abb
...
@@ -7,55 +7,9 @@ import * as fs from 'fs';
...
@@ -7,55 +7,9 @@ import * as fs from 'fs';
import
{
GeneralK8sClient
,
KubernetesCRDClient
}
from
'
../kubernetesApiClient
'
;
import
{
GeneralK8sClient
,
KubernetesCRDClient
}
from
'
../kubernetesApiClient
'
;
import
{
KubeflowOperator
}
from
'
./kubeflowConfig
'
;
import
{
KubeflowOperator
}
from
'
./kubeflowConfig
'
;
/**
* KubeflowOperator Client
*/
abstract
class
KubeflowOperatorClient
extends
KubernetesCRDClient
{
/**
* Factory method to generate operator client
*/
// tslint:disable-next-line:function-name
public
static
generateOperatorClient
(
kubeflowOperator
:
KubeflowOperator
,
operatorApiVersion
:
string
):
KubernetesCRDClient
{
switch
(
kubeflowOperator
)
{
case
'
tf-operator
'
:
{
switch
(
operatorApiVersion
)
{
case
'
v1alpha2
'
:
{
return
new
TFOperatorClientV1Alpha2
();
}
case
'
v1beta1
'
:
{
return
new
TFOperatorClientV1Beta1
();
}
case
'
v1beta2
'
:
{
return
new
TFOperatorClientV1Beta2
();
}
default
:
throw
new
Error
(
`Invalid tf-operator apiVersion
${
operatorApiVersion
}
`
);
}
}
case
'
pytorch-operator
'
:
{
switch
(
operatorApiVersion
)
{
case
'
v1alpha2
'
:
{
return
new
PyTorchOperatorClientV1Alpha2
();
}
case
'
v1beta1
'
:
{
return
new
PyTorchOperatorClientV1Beta1
();
}
case
'
v1beta2
'
:
{
return
new
PyTorchOperatorClientV1Beta2
();
}
default
:
throw
new
Error
(
`Invalid pytorch-operator apiVersion
${
operatorApiVersion
}
`
);
}
}
default
:
throw
new
Error
(
`Invalid operator
${
kubeflowOperator
}
`
);
}
}
}
// tslint:disable: no-unsafe-any no-any completed-docs
// tslint:disable: no-unsafe-any no-any completed-docs
class
TFOperatorClientV1Alpha2
extends
Kube
flowOperator
Client
{
class
TFOperatorClientV1Alpha2
extends
Kube
rnetesCRD
Client
{
/**
/**
* constructor, to initialize tfjob CRD definition
* constructor, to initialize tfjob CRD definition
*/
*/
...
@@ -112,7 +66,7 @@ class TFOperatorClientV1Beta2 extends KubernetesCRDClient {
...
@@ -112,7 +66,7 @@ class TFOperatorClientV1Beta2 extends KubernetesCRDClient {
}
}
}
}
class
PyTorchOperatorClientV1Alpha2
extends
Kube
flowOperator
Client
{
class
PyTorchOperatorClientV1Alpha2
extends
Kube
rnetesCRD
Client
{
/**
/**
* constructor, to initialize tfjob CRD definition
* constructor, to initialize tfjob CRD definition
*/
*/
...
@@ -169,5 +123,51 @@ class PyTorchOperatorClientV1Beta2 extends KubernetesCRDClient {
...
@@ -169,5 +123,51 @@ class PyTorchOperatorClientV1Beta2 extends KubernetesCRDClient {
}
}
}
}
/**
* KubeflowOperator Client
*/
class
KubeflowOperatorClientFactory
{
/**
* Factory method to generate operator client
*/
// tslint:disable-next-line:function-name
public
static
createClient
(
kubeflowOperator
:
KubeflowOperator
,
operatorApiVersion
:
string
):
KubernetesCRDClient
{
switch
(
kubeflowOperator
)
{
case
'
tf-operator
'
:
{
switch
(
operatorApiVersion
)
{
case
'
v1alpha2
'
:
{
return
new
TFOperatorClientV1Alpha2
();
}
case
'
v1beta1
'
:
{
return
new
TFOperatorClientV1Beta1
();
}
case
'
v1beta2
'
:
{
return
new
TFOperatorClientV1Beta2
();
}
default
:
throw
new
Error
(
`Invalid tf-operator apiVersion
${
operatorApiVersion
}
`
);
}
}
case
'
pytorch-operator
'
:
{
switch
(
operatorApiVersion
)
{
case
'
v1alpha2
'
:
{
return
new
PyTorchOperatorClientV1Alpha2
();
}
case
'
v1beta1
'
:
{
return
new
PyTorchOperatorClientV1Beta1
();
}
case
'
v1beta2
'
:
{
return
new
PyTorchOperatorClientV1Beta2
();
}
default
:
throw
new
Error
(
`Invalid pytorch-operator apiVersion
${
operatorApiVersion
}
`
);
}
}
default
:
throw
new
Error
(
`Invalid operator
${
kubeflowOperator
}
`
);
}
}
}
// tslint:enable: no-unsafe-any
// tslint:enable: no-unsafe-any
export
{
KubeflowOperatorClient
,
GeneralK8sClient
};
export
{
KubeflowOperatorClient
Factory
,
GeneralK8sClient
};
src/nni_manager/training_service/kubernetes/kubeflow/kubeflowTrainingService.ts
View file @
72d04abb
...
@@ -21,7 +21,7 @@ import { AzureStorageClientUtility } from '../azureStorageClientUtils';
...
@@ -21,7 +21,7 @@ import { AzureStorageClientUtility } from '../azureStorageClientUtils';
import
{
NFSConfig
}
from
'
../kubernetesConfig
'
;
import
{
NFSConfig
}
from
'
../kubernetesConfig
'
;
import
{
KubernetesTrialJobDetail
}
from
'
../kubernetesData
'
;
import
{
KubernetesTrialJobDetail
}
from
'
../kubernetesData
'
;
import
{
KubernetesTrainingService
}
from
'
../kubernetesTrainingService
'
;
import
{
KubernetesTrainingService
}
from
'
../kubernetesTrainingService
'
;
import
{
KubeflowOperatorClient
}
from
'
./kubeflowApiClient
'
;
import
{
KubeflowOperatorClient
Factory
}
from
'
./kubeflowApiClient
'
;
import
{
KubeflowClusterConfig
,
KubeflowClusterConfigAzure
,
KubeflowClusterConfigFactory
,
KubeflowClusterConfigNFS
,
import
{
KubeflowClusterConfig
,
KubeflowClusterConfigAzure
,
KubeflowClusterConfigFactory
,
KubeflowClusterConfigNFS
,
KubeflowTrialConfig
,
KubeflowTrialConfigFactory
,
KubeflowTrialConfigPytorch
,
KubeflowTrialConfigTensorflow
KubeflowTrialConfig
,
KubeflowTrialConfigFactory
,
KubeflowTrialConfigPytorch
,
KubeflowTrialConfigTensorflow
}
from
'
./kubeflowConfig
'
;
}
from
'
./kubeflowConfig
'
;
...
@@ -136,8 +136,8 @@ class KubeflowTrainingService extends KubernetesTrainingService implements Kuber
...
@@ -136,8 +136,8 @@ class KubeflowTrainingService extends KubernetesTrainingService implements Kuber
nfsKubeflowClusterConfig
.
nfs
.
path
nfsKubeflowClusterConfig
.
nfs
.
path
);
);
}
}
this
.
kubernetesCRDClient
=
KubeflowOperatorClient
.
generateOperatorClient
(
this
.
kubeflowClusterConfig
.
operator
,
this
.
kubernetesCRDClient
=
KubeflowOperatorClient
Factory
.
createClient
(
this
.
kubeflowClusterConfig
.
apiVersion
);
this
.
kubeflowClusterConfig
.
operator
,
this
.
kubeflowClusterConfig
.
apiVersion
);
break
;
break
;
case
TrialConfigMetadataKey
.
TRIAL_CONFIG
:
case
TrialConfigMetadataKey
.
TRIAL_CONFIG
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment