api-ci.yml

name: Run API unit tests
on:
  [push, pull_request]
jobs:
  run-tests:
    runs-on: ubuntu-latest
    steps:
    - name: Checkout
      uses: actions/checkout@v2
    - uses: actions/setup-python@v4
      with:
        python-version: '3.9' 
    - name: install
      run: |
        echo `python3 --version`
        cd torch_ort
        python3 -m pip install --upgrade protobuf==3.20.2
        python3 -m pip install --pre onnxruntime-training --no-cache-dir -f https://download.onnxruntime.ai/onnxruntime_nightly_cpu.html
        python3 -m pip install torch==1.13.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
        python3 setup.py install --prefix=/tmp
        python3 -m torch_ort.configure
        python3 -m pip install pytest
    - name: Print test message
      run: echo Running torch-ort API unit tests
    - name: Run tests
      run: |
        echo `python3 --version`
        cd torch_ort
        python -m pytest -sv tests/torch-ort_test_api.py
        python -m pytest -sv tests/experimental/torch_ort_graph_config_tests.py
        python -m pytest -sv tests/experimental/torch_ort_json_config_tests.py