Fixed axpy in docker build (#134)

NervanaSystems · Feb 6, 2019 · cfd9712 · cfd9712
1 parent 5715261
commit cfd9712
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/build_and_test_he_transformer.sh b/docker/build_and_test_he_transformer.sh
@@ -50,8 +50,8 @@ python_integration()
     echo 'Running python examples'
     cd examples
     python axpy.py
-    NG_TF_BACKEND=HE_SEAL_CKKS python axpy.py
-    NG_TF_BACKEND=HE_SEAL_BFV python axpy.py
+    NGRAPH_TF_BACKEND=HE_SEAL_CKKS python axpy.py
+    NGRAPH_TF_BACKEND=HE_SEAL_BFV python axpy.py
     cd MNIST-Cryptonets
     NGRAPH_HE_SEAL_CONFIG=../../test/model/he_seal_ckks_config_13.json \
         NGRAPH_TF_BACKEND=HE_SEAL_CKKS \