tests.yml

# This workflow will install Python dependencies, run tests and lint with a single version of Python
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

name: Linting, formatting, and tests

on:
  push:
    branches: [ main ]
  pull_request:
    branches: [ main ]

jobs:
  build:
    runs-on: ubuntu-latest

    env:
      PY_VERSION: 3.8
      GPT_ONNX_COMMIT: bb0d4cf3d4e2a5f7376c13a08d337e86296edbe8
      GPT_ONNX_FNAME: gpt2-10.onnx

    steps:
    - uses: actions/checkout@v2

    - name: Set up Python
      uses: actions/setup-python@v2
      with:
        python-version: ${{ env.PY_VERSION }}

    - name: Install pip (and get cache dir)
      id: cache_pip
      run: |
        python -m pip install --upgrade pip
        echo "::set-output name=dir::$(pip cache dir)"

    - name: Cache pip
      uses: actions/cache@v2
      with:
        path: ${{ steps.cache_pip.outputs.dir }}
        key: ${{ runner.os }}-pip-${{ env.PY_VERSION }}-${{ hashFiles('**/requirements.txt') }}
        restore-keys: |
          ${{ runner.os }}-pip-${{ env.PY_VERSION }}
          ${{ runner.os }}-pip-

    - name: Install dependencies
      run: |
        pip install pylint pytest black
        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi

    - name: Cache GPT ONNX file (needed for tests)
      uses: actions/cache@v2
      with:
        path: ${{ env.GPT_ONNX_FNAME }}
        key: ${{ env.GPT_ONNX_COMMIT }}

    - name: Download GPT ONNX file (if not already present)
      run: |
        if [ ! -f "$GPT_ONNX_FNAME" ]; then \
          pushd /tmp; \
          git clone https://github.com/onnx/models.git; \
          pushd models; \
          yes | git lfs migrate import --fixup --everything; \
          git checkout ${{ env.GPT_ONNX_COMMIT }}; \
          git lfs pull --include="text/machine_comprehension/gpt-2/model/${GPT_ONNX_FNAME}" --exclude ""; \
          popd; popd; \
          mv /tmp/models/text/machine_comprehension/gpt-2/model/${GPT_ONNX_FNAME} ./; \
        fi

    - name: Check formatting (black)
      run: black --diff --check .

    - name: Install dist_ir
      run: python setup.py install

    - name: Test with pytest
      run: python -m pytest