setup for tf load_data (#2360)

NVIDIA · Feb 8, 2024 · 22405a3 · 22405a3
1 parent 1bc7196
commit 22405a3
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -22,3 +22,5 @@ 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()"