fix bug

PaddlePaddle · May 8, 2024 · 471d06e · 471d06e
1 parent 4d19ecf
commit 471d06e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
@@ -27,4 +27,5 @@ jobs:
         pip install -e .
     - name: Test with pytest
       run: |
+        lscpu
         pytest tests/
diff --git a/tests/test_paddleocr_api.py b/tests/test_paddleocr_api.py
@@ -23,7 +23,7 @@ def test_ocr_det_rec_api(ocr, img, det, rec):
 
 
 def test_ocr_fp16():
-    ocr = PaddleOCR(lang="ch", use_angle_cls=True, precision=True)
+    ocr = PaddleOCR(lang="ch", use_angle_cls=True, precision="fp16")
     img = cv2.imread("doc/imgs/1.jpg")
     result = ocr.ocr(img, det=True, rec=True)
     assert result is not None