ソースを参照

test: fix assertion and path

Sidney233 3 ヶ月 前
コミット
60dd005dd5
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 = []
@@ -108,7 +108,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"] = "local"
+    # os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
 
     pdf_file_names = []
     pdf_bytes_list = []