test.yml

name: Python package

on: [push, pull_request]

jobs:
  build:

    runs-on: ubuntu-latest

    strategy:
      matrix:
        python-version: [3.8, 3.9, 3.10]

    steps:
    - name: Checkout code
      uses: actions/checkout@v2

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

    - name: Install system dependencies
      run: |
        sudo apt-get update
        sudo apt-get install -y ffmpeg

    - name: Install CUDA Toolkit
      uses: Jimver/cuda-toolkit@v0.2.16
      id: cuda-toolkit
      with:
        cuda: '12.5.0'

    - run: echo "Installed CUDA version is: ${{steps.cuda-toolkit.outputs.cuda}}"

    - run: echo "CUDA install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"

    - run: nvcc -V

    - name: Install Python dependencies
      run: |
        python -m pip install --upgrade pip
        pip install numpy
        pip install cupy-cuda12x
        pip install pydub
        pip install soundfile
        pip install mutagen

    - name: Install package
      run: |
        pip install .

    - name: Run tests
      run: |
        python -m unittest discover -s fat_llama/tests