chore: update CI #15
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
| # This is a basic workflow to help you get started with Actions | |
| name: Run tests | |
| # Controls when the workflow will run | |
| on: | |
| # Triggers the workflow on push or pull request events but only for the main branch | |
| push: | |
| branches: [ main ] | |
| pull_request: | |
| branches: [ main ] | |
| # Allows you to run this workflow manually from the Actions tab | |
| workflow_dispatch: | |
| jobs: | |
| tests: | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| os: [ubuntu-latest] | |
| runs-on: ${{ matrix.os }} | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| - name: Set up Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.9 | |
| - name: Install system dependencies | |
| run: | | |
| sudo apt-get update | |
| sudo apt-get install -y build-essential | |
| sudo apt-get install -y cuda-toolkit-11.2 # Adjust based on your CUDA version | |
| sudo apt-get install -y libomp-dev # Optional for parallel builds | |
| - name: Install Python dependencies | |
| run: | | |
| # Install numpy before torch to avoid missing numpy error | |
| pip install numpy | |
| # Install PyTorch with the specific version (with CUDA 11.2 support here) | |
| pip install torch==2.4.0+cu121 # Modify for the correct CUDA version | |
| # Install packaging, wheel, and other required libraries | |
| pip install packaging wheel | |
| # Install flash-attn with no build isolation | |
| pip install flash-attn==2.6.3 --no-build-isolation | |
| # Install the current package (from setup.py) | |
| pip install . | |
| # Install dev dependencies for testing | |
| pip install -r requirements-dev.txt | |
| # Hacky way to get package name from setup.py | |
| - name: Get package name | |
| id: get_package_name | |
| run: | | |
| setup_file="setup.py" | |
| pattern="name=" | |
| project_name=$(grep -oP "(?<=${pattern}\")[^\"]+" $setup_file) | |
| echo "::set-output name=package_name::$project_name" | |
| - name: Run tests | |
| run: | | |
| pytest -v --cov-fail-under=60 --cov=${{ steps.get_package_name.outputs.package_name }} -l --tb=short --maxfail=1 tests/ | |
| coverage xml | |
| coverage html |