format.yml

# Copyright The FMS HF Tuning Authors
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

name: Format

on:
  push:
    branches: [ "main", "dev" ]
  pull_request:
    branches: [ "main", "dev" ]

jobs:
  lint:
    runs-on: ubuntu-latest
    strategy:
      matrix:
        plugin_name:
          - "framework"
          - "accelerated-peft"
          - "fused-ops-and-kernels"
          - "attention-and-distributed-packing"

    steps:
      - uses: actions/checkout@v4
      - name: Set up Python 3.9
        uses: actions/setup-python@v4
        with:
          python-version: 3.9
      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          python -m pip install tox
      - name: Run linter
        run: |
          cd plugins/${{ matrix.plugin_name }}
          tox -e lint
      - name: Run formatter
        run: |
          cd plugins/${{ matrix.plugin_name }}
          tox -e fmt -- . --check
      - name: Run pytest
        run: |
          cd plugins/${{ matrix.plugin_name }}
          tox -e py

  sample-config:
    runs-on: ubuntu-latest

    steps:
      - uses: actions/checkout@v4
      - name: Set up Python 3.9
        uses: actions/setup-python@v4
        with:
          python-version: 3.9
      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          python -m pip install tox
      - name: Run Config Verification
        run: tox -e verify-configs