From 804bd924f2e4d4b7d9a96736a58c78b79f1a2ed8 Mon Sep 17 00:00:00 2001 From: Janusz Lisiecki <39967756+JanuszL@users.noreply.github.com> Date: Fri, 17 Jun 2022 20:06:00 +0200 Subject: [PATCH] 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 --- dali/test/python/test_dali_tf_dataset_mnist_eager.py | 4 ++-- 1 file changed, 2 insertions(+), 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 index 8424bf8a5c8..3c479ba4269 100644 --- 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()