File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change 8
8
from fastdeploy .engine .request import RequestOutput
9
9
from fastdeploy .engine .sampling_params import SamplingParams
10
10
from fastdeploy .entrypoints .llm import LLM
11
- from fastdeploy .utils import get_random_port
12
11
13
12
MODEL_NAME = os .getenv ("MODEL_PATH" ) + "/ernie-45-21b-a3b-bf16-paddle"
14
13
@@ -32,7 +31,6 @@ class TestGeneration(unittest.TestCase):
32
31
33
32
@classmethod
34
33
def setUpClass (cls ):
35
- """Set up test environment before any tests run"""
36
34
try :
37
35
llm = LLM (
38
36
model = MODEL_NAME ,
@@ -42,7 +40,8 @@ def setUpClass(cls):
42
40
)
43
41
cls .llm = weakref .proxy (llm )
44
42
except Exception as e :
45
- return
43
+ print (f"Setting up LLM failed: { e } " )
44
+ raise unittest .SkipTest (f"LLM initialization failed: { e } " )
46
45
47
46
@classmethod
48
47
def tearDownClass (cls ):
You can’t perform that action at this time.
0 commit comments