Skip to content

Update requirements, add bench script #1

Update requirements, add bench script

Update requirements, add bench script #1

Workflow file for this run

name: Integration test
on: [push]
env:
TORCH_DEVICE: "cpu"
jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- name: Set up Python 3.11
uses: actions/setup-python@v4
with:
python-version: 3.11
- name: Install apt dependencies
run: |
sudo apt-get update
sudo apt-get install -y tesseract-ocr tesseract-ocr-eng
- name: Install python dependencies
run: |
pip install poetry
poetry install
poetry remove torch
poetry run pip install torch --index-url https://download.pytorch.org/whl/cpu
- name: Run benchmark test
run: |
poetry run python benchmarks/benchmark.py --max 5 temp.json
poetry run python scripts/verify_benchmark_scores.py temp.json