fixingray init script

Author: webcoderz

modules/api/ray.py

@@ -14,7 +14,20 @@
 
 def ray_only():
     serve.shutdown()
-    
+
+    if "RAY_DOCKER" in os.environ:
+        ray.init(
+            dashboard_host=os.environ.get("RAY_DASHBOARD_HOST", "0.0.0.0"),
+            dashboard_port=int(os.environ.get("RAY_DASHBOARD_PORT", 8265))
+            )
+    elif "RAY_HEAD_ADDRESS" in os.environ:
+        ray.init(address=os.environ.get("RAY_HEAD_ADDRESS", ""))
+    elif "RAY_SERVE_ONLY" in os.environ:
+        pass
+    else:
+        ray.init()
+
+
     print("starting ray")
     serve.start(
 
@@ -29,18 +42,3 @@ def ray_only():
     print("Done setting up replicas! Now accepting requests...")
     while True:
         time.sleep(1000)
-
-
-if __name__ == "__main__":
-
-    if "RAY_DOCKER" in os.environ:
-        ray.init(
-            dashboard_host=os.environ.get("RAY_DASHBOARD_HOST", "0.0.0.0"),
-            dashboard_port=int(os.environ.get("RAY_DASHBOARD_PORT", 8265))
-            )
-    elif "RAY_HEAD_ADDRESS" in os.environ:
-        ray.init(address=os.environ.get("RAY_HEAD_ADDRESS", ""))
-    elif "RAY_SERVE_ONLY" in os.environ:
-        ray_only()
-    else:
-        ray.init()