revert auto init hook

Signed-off-by: Eric Liang <ekhliang@gmail.com>
ray-project · Nov 30, 2023 · 53cfcbb · 53cfcbb
1 parent a019d22
commit 53cfcbb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/ray/_private/auto_init_hook.py b/python/ray/_private/auto_init_hook.py
@@ -4,11 +4,13 @@
 import threading
 
 auto_init_lock = threading.Lock()
-enable_auto_connect = os.environ.get("RAY_ENABLE_AUTO_CONNECT", "") != "0"
 
 
 def auto_init_ray():
-    if enable_auto_connect and not ray.is_initialized():
+    if (
+        os.environ.get("RAY_ENABLE_AUTO_CONNECT", "") != "0"
+        and not ray.is_initialized()
+    ):
         auto_init_lock.acquire()
         if not ray.is_initialized():
             ray.init()