Bläddra i källkod

test: update e2e

Sidney233 4 månader sedan
förälder
incheckning
fc2377ff37
1 ändrade filer med 2 tillägg och 2 borttagningar
  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"] = "modelscope"
+    # os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
 
     pdf_file_names = []
     pdf_bytes_list = []
@@ -102,7 +102,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 = []