Skip to content

Commit

Permalink
added github workflow
Browse files Browse the repository at this point in the history
  • Loading branch information
theramkm committed Dec 7, 2024
1 parent 40b032b commit cad3f12
Showing 1 changed file with 30 additions and 0 deletions.
30 changes: 30 additions & 0 deletions .github/workflows/model_tests.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
name: Model Architecture Tests

on:
push:
branches: [ main ]
pull_request:
branches: [ main ]

jobs:
test:
runs-on: ubuntu-latest

steps:
- uses: actions/checkout@v4

- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: '3.10'

- name: Install dependencies
run: |
python -m pip install --upgrade pip
# Install CPU-only version of PyTorch
pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
pip install pytest
- name: Run tests
run: |
pytest tests/test_model.py -v

0 comments on commit cad3f12

Please sign in to comment.