Fix pytest_sessionstart

AlibabaPAI · Feb 20, 2025 · 3cad314 · 3cad314
1 parent 862e067
commit 3cad314
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/llumnix/entrypoints/setup.py b/llumnix/entrypoints/setup.py
@@ -16,6 +16,7 @@
 import os
 import time
 from typing import Dict, Optional, List, Tuple
+import traceback
 import ray
 
 from llumnix.manager import Manager

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -30,6 +30,10 @@
 from llumnix.utils import random_uuid
 
 
+def pytest_sessionstart(session):
+    subprocess.run(["ray", "start", "--head", "--disable-usage-stats", "--port=6379"], check=False,
+                   stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
 def cleanup_ray_env_func():
     try:
         # list_placement_groups cannot take effects.