File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 4
4
variables :
5
5
FRAMEWORK_VERSION : ' 1.13.1'
6
6
GPU_INSTANCE_TYPE : ' ml.p2.xlarge'
7
+ SETUP_FILE : ' setup_cmds.sh'
8
+ SETUP_CMDS : ' #!/bin/bash\npip install --upgrade pip\npip install -U -e .\npip install -U -e .[test]'
7
9
8
10
phases :
9
11
pre_build :
@@ -45,11 +47,12 @@ phases:
45
47
- instance_type=${GPU_INSTANCE_TYPE#"$prefix"}
46
48
- create-key-pair
47
49
- launch-ec2-instance --instance-type $instance_type --ami-name dlami-ubuntu
50
+ - printf "$SETUP_CMDS" > $SETUP_FILE
48
51
49
52
- py2_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name $base_name --py-version 2 --framework-version $FRAMEWORK_VERSION --processor gpu"
50
- - remote-test --github-repo sagemaker-tensorflow-container --branch script-mode --test-cmd "$py2_cmd"
53
+ - remote-test --github-repo sagemaker-tensorflow-container --setup-file $SETUP_FILE -- branch script-mode --test-cmd "$py2_cmd"
51
54
- py3_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name $base_name --framework-version $FRAMEWORK_VERSION --processor gpu"
52
- - remote-test --github-repo sagemaker-tensorflow-container --branch script-mode --test-cmd "$py3_cmd"
55
+ - remote-test --github-repo sagemaker-tensorflow-container --setup-file $SETUP_FILE -- branch script-mode --test-cmd "$py3_cmd"
53
56
54
57
- |
55
58
echo '[{
You can’t perform that action at this time.
0 commit comments