update comment

intel-analytics · Mar 26, 2024 · 35913c8 · 35913c8
1 parent a82b0cb
commit 35913c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/community/tests/integration_tests/llms/test_bigdl_llm.py b/libs/community/tests/integration_tests/llms/test_bigdl_llm.py
@@ -5,7 +5,7 @@
 
 
 def test_call() -> None:
-    """Test valid call to ipex-llm."""
+    """Test valid call to bigdl-llm."""
     llm = BigdlLLM.from_model_id(
         model_id="lmsys/vicuna-7b-v1.5",
         model_kwargs={"temperature": 0, "max_length": 16, "trust_remote_code": True},
@@ -15,7 +15,7 @@ def test_call() -> None:
 
 
 def test_generate() -> None:
-    """Test valid call to ipex-llm."""
+    """Test valid call to bigdl-llm."""
     llm = BigdlLLM.from_model_id(
         model_id="lmsys/vicuna-7b-v1.5",
         model_kwargs={"temperature": 0, "max_length": 16, "trust_remote_code": True},