main.yml

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


jobs:
  pytest:
    runs-on: ${{ matrix.os }}
    strategy:
      matrix:
        os: [ ubuntu-latest, macos-latest ]
        python-version: [3.8, 3.9, "3.10"]
        include:
          - os: ubuntu-latest
            path: ~/.cache/pip
          - os: macos-latest
            path: ~/Library/Caches/pip
    env:
      OS: ${{ matrix.os }}
      PYTHON: '3.10'


    steps:
      - uses: actions/checkout@v2
        with:
          fetch-depth: 0  # Shallow clones should be disabled for a better relevancy of analysis

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

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

      - name: Install dependencies
        env:
            TORCH_URL: https://download.pytorch.org/whl/cpu/torch_stable.html
        run: |
          python --version
          pip --version
          python -m pip install --upgrade pip coverage pytest
          pip install --index-url https://download.pytorch.org/whl/nightly/cpu --pre 'torch>=2.1.0dev'
          pip install lit_gpt@git+https://github.com/aniketmaurya/install-lit-gpt.git@install
          pip install . --quiet
          pip list
        shell: bash

      - name: Prepare Test
        run: |
          python tests  # download test data

      - name: Run Test with Coverage
        run: |
          coverage erase
          coverage run -m pytest

      - name: Generate Coverage Report
        run: |
          coverage report -m -i
          coverage xml -i

      - name: Upload Coverage to Codecov
        if: runner.os != 'macOS'
        uses: codecov/codecov-action@v1
        with:
          token: ${{ secrets.CODECOV_TOKEN }}
          file: ./coverage.xml
          flags: unittests
          env_vars: OS,PYTHON
          name: codecov-umbrella
          fail_ci_if_error: false