From 47c51afad7da83b42d637ca81bbf5b94456a8d76 Mon Sep 17 00:00:00 2001 From: Jonathan Chang Date: Mon, 9 Dec 2024 11:27:07 -0800 Subject: [PATCH] Ensure torch install on CI --- .github/workflows/ocr-benchmarks.yml | 4 ++-- .github/workflows/ocr-tests.yml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/ocr-benchmarks.yml b/.github/workflows/ocr-benchmarks.yml index fcafb89a..2b68b18d 100644 --- a/.github/workflows/ocr-benchmarks.yml +++ b/.github/workflows/ocr-benchmarks.yml @@ -20,7 +20,7 @@ jobs: python -m pip install --upgrade pip pip install poetry pip install pytest pytest-benchmark - poetry install --with dev + poetry install --with dev --with torch - name: Run tests run: | cd OCR/tests @@ -43,4 +43,4 @@ jobs: # cp ./dev/bench/index-template.html ./dev/bench/index.html # git add . # git commit -m "template to index" - # git push origin gh-pages \ No newline at end of file + # git push origin gh-pages diff --git a/.github/workflows/ocr-tests.yml b/.github/workflows/ocr-tests.yml index c3754eee..abc703c1 100644 --- a/.github/workflows/ocr-tests.yml +++ b/.github/workflows/ocr-tests.yml @@ -20,6 +20,6 @@ jobs: sudo apt install tesseract-ocr-eng tesseract-ocr -y python -m pip install --upgrade pip pip install poetry - poetry install --with dev + poetry install --with dev --with torch - name: Run tests run: poetry run pytest tests/ocr_test.py tests/segmentation_template_test.py tests/alignment_test.py