From 76f6a815ff5b936ea5edc2879e8ce088d4efd8c9 Mon Sep 17 00:00:00 2001 From: Alistair Muldal Date: Tue, 13 Apr 2021 17:07:51 +0100 Subject: [PATCH] Enable Travis tests for `adversarial_robustness` * Ensure that `pip` is up to date * Remove invalid `arch` flag PiperOrigin-RevId: 368226929 --- .travis.yml | 1 + adversarial_robustness/run.sh | 7 +++---- 2 files changed, 4 insertions(+), 4 deletions(-) mode change 100644 => 100755 adversarial_robustness/run.sh diff --git a/.travis.yml b/.travis.yml index 8074d1e..1ddb877 100644 --- a/.travis.yml +++ b/.travis.yml @@ -8,6 +8,7 @@ python: - "3.6" env: matrix: + - PROJECT="adversarial_robustness" - PROJECT="avae" # - PROJECT="cs_gan" # TODO(b/184845450): Fix and re-enable - PROJECT="iodine" diff --git a/adversarial_robustness/run.sh b/adversarial_robustness/run.sh old mode 100644 new mode 100755 index e901ed5..0b64b41 --- a/adversarial_robustness/run.sh +++ b/adversarial_robustness/run.sh @@ -13,20 +13,19 @@ # See the License for the specific language governing permissions and # limitations under the License. -python3 -m venv adversarial_robustness_venv -source adversarial_robustness_venv/bin/activate +python3 -m venv /tmp/adversarial_robustness_venv +source /tmp/adversarial_robustness_venv/bin/activate +pip install -U pip pip install -r adversarial_robustness/requirements.txt python3 -m adversarial_robustness.jax.eval \ --ckpt=dummy \ - --arch=wrn-10-1 \ --dataset=cifar10 \ --batch_size=1 \ --num_batches=1 python3 -m adversarial_robustness.pytorch.eval \ --ckpt=dummy \ - --arch=wrn-10-1 \ --dataset=cifar10 \ --batch_size=1 \ --num_batches=1 \