Browse Source

test: fix assertion and path

Sidney233 3 months ago
parent
commit
60dd005dd5
1 changed files with 2 additions and 2 deletions
  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 = []