Skip to content

Commit

Permalink
add a shark ci
Browse files Browse the repository at this point in the history
  • Loading branch information
dan-garvey committed Dec 20, 2023
1 parent 68df316 commit 1b48f1b
Showing 1 changed file with 52 additions and 0 deletions.
52 changes: 52 additions & 0 deletions .github/workflows/test_shark.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: Test SHARK

on:
workflow_dispatch:
pull_request:
push:
branches:
- main

concurrency:
# A PR number if a pull request and otherwise the commit hash. This cancels
# queued and in-progress runs for the same PR (presubmit) or commit
# (postsubmit). The workflow name is prepended to avoid conflicts between
# different workflows.
group: ${{ github.workflow }}-${{ github.event.number || github.sha }}
cancel-in-progress: true

jobs:
test-shark:
strategy:
matrix:
version: [3.11]
os: [nodai-ubuntu-builder-large]

runs-on: ${{matrix.os}}
steps:
- name: "Setting up Python"
uses: actions/setup-python@75f3110429a8c05be0e1bf360334e4cced2b63fa # v2.3.3
with:
python-version: ${{matrix.version}}

- name: "Checkout SHARK"
uses: actions/checkout@v4
with:
repository: "nod-ai/SHARK.git"
path: SHARK
ref: "main"

# TODO: Replace with a sh script from shark repo
- name: "Install SHARK"
run: |
cd $GITHUB_WORKSPACE/SHARK
python${{ matrix.version }} -m venv shark.venv
source shark.venv/bin/activate
pip install -r requirements.txt --no-cache-dir
pip install -e .
pip uninstall -y torch
pip install torch==2.1.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
pip uninstall shark-turbine turbine-models
pip install -y "shark-turbine @ git+https://github.com/nod-ai/SHARK-Turbine.git@${{ github.sha }}"
pip install -y "turbine-models @ git+https://github.com/nod-ai/SHARK-Turbine@${{ github.sha }}#egg=turbine-models&subdirectory=python/turbine_models"
python apps/shark_studio/tests/api_test.py

0 comments on commit 1b48f1b

Please sign in to comment.