diff --git a/tests/utils/test_device.py b/tests/utils/test_device.py index 3135b214..4def9915 100644 --- a/tests/utils/test_device.py +++ b/tests/utils/test_device.py @@ -81,6 +81,7 @@ class DeviceTest(unittest.TestCase): with device_placement(Frameworks.torch, 'cpu'): pass + @unittest.skip('skip this test to avoid debug logging.') def test_device_placement_tf_gpu(self): tf.debugging.set_log_device_placement(True) with device_placement(Frameworks.tf, 'gpu:0'):