tf setup and teardown for integration tests (#2366)

NVIDIA · Feb 16, 2024 · 5341fc1 · 5341fc1
1 parent 22405a3
commit 5341fc1
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/examples/nvflare_setup.ipynb b/examples/nvflare_setup.ipynb
@@ -23,7 +23,7 @@
    "source": [
     "### Install NVFLARE from PyPI\n",
     "```\n",
-    "pip install 'nvflare>=2.3.0'\n",
+    "pip install 'nvflare~=2.4.0'\n",
     "\n",
     "```\n",
     "We do not recommend running NVFlare CLI commands in jupyter notebook cells."

diff --git a/tests/integration_test/data/test_configs/standalone_job/hello_tf_examples.yml b/tests/integration_test/data/test_configs/standalone_job/hello_tf_examples.yml
@@ -22,6 +22,10 @@ tests:
           "data": { "run_finished": True }
     validators:
       - path: tests.integration_test.src.validators.TFModelValidator
+    setup:
+      - python -c "import tensorflow as tf; tf.keras.datasets.mnist.load_data()"
+    teardown:
+      - rm ~/.keras/datasets/mnist.npz
   - test_name: "run hello-tf2"
     event_sequence:
       - "trigger":
@@ -39,3 +43,7 @@ tests:
           "data": { "run_finished": True }
     validators:
       - path: tests.integration_test.src.validators.TFModelValidator
+    setup:
+      - python -c "import tensorflow as tf; tf.keras.datasets.mnist.load_data()"
+    teardown:
+      - rm ~/.keras/datasets/mnist.npz
diff --git a/tests/integration_test/data/test_configs/standalone_job/internal_tf.yml b/tests/integration_test/data/test_configs/standalone_job/internal_tf.yml
@@ -24,3 +24,5 @@ tests:
       - path: tests.integration_test.src.validators.TFModelValidator
     setup:
       - python -c "import tensorflow as tf; tf.keras.datasets.mnist.load_data()"
+    teardown:
+      - rm ~/.keras/datasets/mnist.npz