Skip to content

adding support for raw ONNX models #7

adding support for raw ONNX models

adding support for raw ONNX models #7

name: Run Torch 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/torch_assets.py
deactivate
- name: Run Python Unit Tests
run: |
source venv/bin/activate
export PYTHONPATH="."
python tests/unit_tests/engine/test_torch.py
python tests/unit_tests/test_rust_adapter.py
python tests/unit_tests/test_surml_file.py
deactivate
- name: Run Core Unit Tests
run: cd modules/core && cargo test --features torch-tests