Fix test_dali_tf_dataset_mnist_eager test (#3991)

- fixes missing import of available_gpus in the test_dali_tf_dataset_mnist_eager test Signed-off-by: Janusz Lisiecki <[email protected]>
NVIDIA · Jun 17, 2022 · 804bd92 · 804bd92
1 parent 7238f15
commit 804bd92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dali/test/python/test_dali_tf_dataset_mnist_eager.py b/dali/test/python/test_dali_tf_dataset_mnist_eager.py
@@ -16,7 +16,7 @@
 from nose.tools import with_setup
 
 import test_dali_tf_dataset_mnist as mnist
-from test_utils_tensorflow import skip_for_incompatible_tf
+from test_utils_tensorflow import skip_for_incompatible_tf, available_gpus
 from nose_utils import raises
 
 tf.compat.v1.enable_eager_execution()
@@ -62,7 +62,7 @@ def test_keras_wrong_placement_cpu():
 
 @with_setup(skip_for_incompatible_tf)
 def test_keras_multi_gpu_mirrored_strategy():
-    strategy = tf.distribute.MirroredStrategy(devices=mnist.available_gpus())
+    strategy = tf.distribute.MirroredStrategy(devices=available_gpus())
 
     with strategy.scope():
         model = mnist.keras_model()