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
6f272049
"src/targets/gpu/vscode:/vscode.git/clone" did not exist on "53ae18eae81cffabc315afbbc0655e71d8ae8efe"
Unverified
Commit
6f272049
authored
Jan 26, 2021
by
SparkSnail
Committed by
GitHub
Jan 26, 2021
Browse files
Remove paiYarn mode (#3327)
parent
f11aea0d
Changes
21
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
84 deletions
+0
-84
ts/nni_manager/training_service/test/paiYarnTrainingService.test.ts
...ager/training_service/test/paiYarnTrainingService.test.ts
+0
-84
No files found.
ts/nni_manager/training_service/test/paiYarnTrainingService.test.ts
deleted
100644 → 0
View file @
f11aea0d
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.
'
use strict
'
;
import
*
as
chai
from
'
chai
'
;
import
*
as
chaiAsPromised
from
'
chai-as-promised
'
;
import
*
as
fs
from
'
fs
'
;
import
*
as
tmp
from
'
tmp
'
;
import
*
as
component
from
'
../../common/component
'
;
import
{
TrialJobApplicationForm
}
from
'
../../common/trainingService
'
;
import
{
cleanupUnitTest
,
prepareUnitTest
}
from
'
../../common/utils
'
;
import
{
TrialConfigMetadataKey
}
from
'
../common/trialConfigMetadataKey
'
;
import
{
PAIYarnTrainingService
}
from
'
../pai/paiYarn/paiYarnTrainingService
'
;
// TODO: copy mockedTrail.py to local folder
const
localCodeDir
:
string
=
tmp
.
dirSync
().
name
const
mockedTrialPath
:
string
=
'
./training_service/test/mockedTrial.py
'
fs
.
copyFileSync
(
mockedTrialPath
,
localCodeDir
+
'
/mockedTrial.py
'
)
describe
(
'
Unit Test for PAIYarnTrainingService
'
,
()
=>
{
let
skip
:
boolean
=
false
;
let
testPaiClusterInfo
:
any
;
let
paiCluster
:
any
;
let
paiTrialConfig
:
any
;
try
{
testPaiClusterInfo
=
JSON
.
parse
(
fs
.
readFileSync
(
'
../../.vscode/paiCluster.json
'
,
'
utf8
'
));
paiCluster
=
`{\"userName\":\"
${
testPaiClusterInfo
.
userName
}
\",\"passWord\":\"
${
testPaiClusterInfo
.
passWord
}
\",\"host\":\"
${
testPaiClusterInfo
.
host
}
\"}`
;
paiTrialConfig
=
`{\"command\":\"echo hello && ls\",\"codeDir\":\"/tmp/nni/examples/trials/mnist",\"gpuNum\":\"1\",
\"cpuNum\":\"1\",\"memoryMB\":\"8196\",\"image\":\"openpai/pai.example.tensorflow\",\"dataDir\":\"\",\"outputDir\":\"\"}`
;
}
catch
(
err
)
{
console
.
log
(
'
Please configure rminfo.json to enable remote machine unit test.
'
);
skip
=
true
;
}
let
paiYarnTrainingService
:
PAIYarnTrainingService
;
console
.
log
(
tmp
.
dirSync
().
name
);
before
(()
=>
{
chai
.
should
();
chai
.
use
(
chaiAsPromised
);
prepareUnitTest
();
});
after
(()
=>
{
cleanupUnitTest
();
});
beforeEach
(()
=>
{
if
(
skip
)
{
return
;
}
paiYarnTrainingService
=
component
.
get
(
PAIYarnTrainingService
);
paiYarnTrainingService
.
run
();
});
afterEach
(()
=>
{
if
(
skip
)
{
return
;
}
paiYarnTrainingService
.
cleanUp
();
});
it
(
'
Get PAI token
'
,
async
()
=>
{
if
(
skip
)
{
return
;
}
console
.
log
(
`paiCluster is
${
paiCluster
}
`
)
await
paiYarnTrainingService
.
setClusterMetadata
(
TrialConfigMetadataKey
.
PAI_YARN_CLUSTER_CONFIG
,
paiCluster
);
await
paiYarnTrainingService
.
setClusterMetadata
(
TrialConfigMetadataKey
.
TRIAL_CONFIG
,
paiTrialConfig
);
const
form
:
TrialJobApplicationForm
=
{
sequenceId
:
0
,
hyperParameters
:
{
value
:
''
,
index
:
0
}
};
try
{
const
trialDetail
=
await
paiYarnTrainingService
.
submitTrialJob
(
form
);
chai
.
expect
(
trialDetail
.
status
).
to
.
be
.
equals
(
'
WAITING
'
);
}
catch
(
error
)
{
console
.
log
(
'
Submit job failed:
'
+
error
);
chai
.
assert
(
error
)
}
});
});
Prev
1
2
Next
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