Skip to content

Commit 21a5bfe

Browse files
alimuldaldiegolascasas
authored andcommitted
Enable Travis tests for hierarchical_probabilistic_unet
* Fix relative paths in `run.sh` so that it can be invoked from the parent directory * Ensure that `pip` is up to date * Don't explicitly `deactivate` - this causes issues since `run.sh` gets executed from within another `virtualenv` created by Travis PiperOrigin-RevId: 368236037
1 parent 20e06f1 commit 21a5bfe

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

.travis.yml

+1
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ env:
1313
# - PROJECT="cs_gan" # TODO(b/184845450): Fix and re-enable
1414
- PROJECT="gated_linear_networks"
1515
- PROJECT="geomancer"
16+
- PROJECT="hierarchical_probabilistic_unet"
1617
- PROJECT="iodine"
1718
- PROJECT="kfac_ferminet_alpha"
1819
- PROJECT="learning_to_simulate"

hierarchical_probabilistic_unet/run.sh

100644100755
+5-5
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
1515

16-
python3 -m venv hpu-net-venv
17-
source hpu-net-venv/bin/activate
18-
pip3 install .
19-
python3 model_test.py
20-
deactivate
16+
python3 -m venv /tmp/hpu-net-venv
17+
source /tmp/hpu-net-venv/bin/activate
18+
pip3 install -U pip
19+
pip3 install hierarchical_probabilistic_unet/
20+
python3 hierarchical_probabilistic_unet/model_test.py

0 commit comments

Comments
 (0)