Test orig_devices for not None

horovod · Mar 1, 2022 · 477a95a · 477a95a
1 parent d627a27
commit 477a95a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/single/test_ray.py b/test/single/test_ray.py
@@ -55,7 +55,7 @@ def ray_start_4_cpus_4_gpus():
         # The code after the yield will run as teardown code.
         ray.shutdown()
     finally:
-        if orig_devices:
+        if orig_devices is not None:
             os.environ["CUDA_VISIBLE_DEVICES"] = orig_devices
         else:
             del os.environ["CUDA_VISIBLE_DEVICES"]