Skip to content

Commit 8e97672

Browse files
authored
fix: add setup file back (#196)
1 parent cfc4ac5 commit 8e97672

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

buildspec-release.yml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ env:
44
variables:
55
FRAMEWORK_VERSION: '1.13.1'
66
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]'
79

810
phases:
911
pre_build:
@@ -45,11 +47,12 @@ phases:
4547
- instance_type=${GPU_INSTANCE_TYPE#"$prefix"}
4648
- create-key-pair
4749
- launch-ec2-instance --instance-type $instance_type --ami-name dlami-ubuntu
50+
- printf "$SETUP_CMDS" > $SETUP_FILE
4851

4952
- 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"
5154
- 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"
5356

5457
- |
5558
echo '[{

0 commit comments

Comments
 (0)