Skip to content

Adding Raw ONNX support to SurrealML #4

Adding Raw ONNX support to SurrealML

Adding Raw ONNX support to SurrealML #4

name: Run Onnx Tests
on:
pull_request:
types: [opened, reopened, synchronize]
jobs:
test_core:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
with:
fetch-depth: 0
- name: Set up Rust
uses: actions-rs/toolchain@v1
with:
profile: minimal
toolchain: stable
override: true
- name: Set up Python
uses: actions/setup-python@v2
with:
python-version: '3.11'
- name: Pre-test Setup
run: |
python3 -m venv venv
source venv/bin/activate
pip install --upgrade pip
pip install -r requirements.txt
# build the local version of the core module to be loaded into python
echo "Building local version of core module"
pip install .
export PYTHONPATH="."
python ./tests/scripts/ci_local_build.py
echo "Local build complete"
# train the models for the tests
python ./tests/model_builder/onnx_assets.py
deactivate
- name: Run Core Unit Tests
run: cd modules/core && cargo test --features onnx-tests