Ensure torch install on CI

CDCgov · Dec 9, 2024 · 47c51af · 47c51af
1 parent b506593
commit 47c51af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
+      #     git push origin gh-pages
diff --git 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