updating to ort v2 #18
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |