update

baidubce · Dec 30, 2024 · 1e2c68b · 1e2c68b
1 parent db9fbb2
commit 1e2c68b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/tests/component_tool_eval_cases.py b/python/tests/component_tool_eval_cases.py
@@ -90,7 +90,7 @@ def schemas(self):
 
 class StyleRewriteCase(Case):
     def init_args(self):
-        return {"model": "ERNIE-3.5-8K"}
+        return {"model": "Qianfan-Agent-Speed-8k"}
 
     def inputs(self):
         return {"query": "文心大模型发布新版"}

diff --git a/python/tests/test_v2_component_trace.py b/python/tests/test_v2_component_trace.py
@@ -21,7 +21,7 @@
 @unittest.skipUnless(os.getenv("TEST_CASE", "UNKNOWN") == "CPU_PARALLEL", "")
 class TestStyleRewrite(unittest.TestCase):
     def setUp(self) -> None:
-        self.com = StyleRewrite(model="Qianfan-Agent-Speed-8k")
+        self.com = StyleRewrite(model="ERNIE-3.5-8K")
         self.tracer = AppBuilderTracer(
             enable_phoenix = False,
             enable_console = True

diff --git a/python/tests/test_v2_plant_recognize.py b/python/tests/test_v2_plant_recognize.py
@@ -21,7 +21,7 @@
 from appbuilder.core.component import ComponentOutput
 from appbuilder.core.components.v2.plant_recognize.component import PlantRecognition
 
-# @unittest.skipUnless(os.getenv("TEST_CASE", "UNKNOWN") == "CPU_PARALLEL", "")
+@unittest.skipUnless(os.getenv("TEST_CASE", "UNKNOWN") == "CPU_PARALLEL", "")
 class TestPlantRecognition(unittest.TestCase):
 
     def setUp(self):