-
Notifications
You must be signed in to change notification settings - Fork 720
/
tests_nightly_pypi.sh
executable file
·58 lines (44 loc) · 1.53 KB
/
tests_nightly_pypi.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#!/bin/bash
# Install pypi from test: ./test_nightly_pypi.sh test
# Install pypi from pypi: ./test_nightly_pypi.sh official
# Exit if any process returns non-zero status.
set -e
# Display the commands being run in logs, which are replicated to sponge.
set -x
if [[ $# -lt 1 ]] ; then
echo "Usage:"
echo "test_nightly_pypi [test|official]"
exit 1
fi
run_tests() {
echo "run_tests $1 $2"
TMP=$(mktemp -d)
# Create and activate a virtualenv to specify python version and test in
# isolated environment. Note that we don't actually have to cd'ed into a
# virtualenv directory to use it; we just need to source bin/activate into the
# current shell.
VENV_PATH=${TMP}/virtualenv/$1
virtualenv -p "$1" "${VENV_PATH}"
source ${VENV_PATH}/bin/activate
# TensorFlow isn't a regular dependency because there are many different pip
# packages a user might have installed.
if [[ $2 == "test" ]] ; then
pip install tf-nightly==1.15.0.dev20190821
# Install pypi package from test.pypi
pip install --index-url https://test.pypi.org/simple/ tf-agents-nightly
elif [[ $2 == "official" ]] ; then
pip install tf-nightly==1.15.0.dev20190821
# Install pypi package
pip install tf-agents-nightly
else
echo "Error unknown option only [test|official]"
exit
fi
# Move away from repo directory so "import tf_agents" refers to the
# installed wheel and not to the local fs.
(cd $(mktemp -d) && python -c 'import tf_agents')
# Deactivate virtualenv
deactivate
}
# Test on Python3.6
run_tests "python3.6" $1