浏览代码

test: fix path

Sidney233 3 月之前
父节点
当前提交
0a91a69596
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      tests/unittest/test_e2e.py

+ 2 - 2
tests/unittest/test_e2e.py

@@ -36,7 +36,7 @@ def test_pipeline_with_two_config():
         if doc_path.suffix in pdf_suffixes + image_suffixes:
             doc_path_list.append(doc_path)
 
-    os.environ["MINERU_MODEL_SOURCE"] = "local"
+    # os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
 
     pdf_file_names = []
     pdf_bytes_list = []
@@ -105,7 +105,7 @@ def test_vlm_transformers_with_default_config():
         if doc_path.suffix in pdf_suffixes + image_suffixes:
             doc_path_list.append(doc_path)
 
-    os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
+    # os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
 
     pdf_file_names = []
     pdf_bytes_list = []