Enable Travis tests for nfnets

PiperOrigin-RevId: 368236627
This commit is contained in:
Alistair Muldal
2021-04-13 17:57:11 +01:00
committed by Diego de Las Casas
parent 21a5bfec5b
commit 9c9e41e1fc
2 changed files with 4 additions and 3 deletions

View File

@@ -17,6 +17,7 @@ env:
- PROJECT="iodine"
- PROJECT="kfac_ferminet_alpha"
- PROJECT="learning_to_simulate"
- PROJECT="nfnets"
- PROJECT="noise_covariance"
- PROJECT="option_keyboard"
# - PROJECT="polygen" # TODO(b/184847339): Fix and re-enable

6
nfnets/run.sh Normal file → Executable file
View File

@@ -14,9 +14,9 @@
# limitations under the License.
set -e
python3 -m venv nfnets_venv
source nfnets_venv/bin/activate
pip3 install --upgrade setuptools wheel
python3 -m venv /tmp/nfnets_venv
source /tmp/nfnets_venv/bin/activate
pip3 install --upgrade pip setuptools wheel
pip3 install -r nfnets/requirements.txt
python3 -m nfnets.test