CPUTests.yml

name: Linter

on:
  push:
    branches:
      - '**'

jobs:
  cpu:
    name: "CPU tests"
    runs-on: ${{ matrix.os }}
    strategy:
      matrix:
        os: [ubuntu-20.04]
        python-version: ['3.10']
    steps:
    - uses: actions/checkout@v3
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v4
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install Dependencies
      run: |
        python -m pip install --upgrade pip
        pip install pylint pyink pytype==2024.2.27
    - name: Typecheck the code with pytype
      run: |
        pytype --jobs auto --disable import-error MaxText/
    - name: Analysing the code with pylint in Maxtext/
      run: |
         pylint MaxText/  && \
         echo 'Maxtext PyLint check successful' || { echo \
         'PyLint check has failed. Please run bash code_style.sh to fix issues'; exit 20; }
    - name: Analysing the code with pylint in pedagogical_examples/
      run: |
         pylint pedagogical_examples/ && \
         echo 'PyLint check on pedagogical_examples/ is successful' || { echo \
         'PyLint check has failed. Please run bash code_style.sh to fix issues'; exit 20; }
    - name: Analysing the code with pyink in Maxtext/
      run: |
        pyink MaxText --check --diff --color --pyink-indentation=2 --line-length=125
    - name: Analysing the code with pyink in pedagogical_examples/
      run: |
        pyink pedagogical_examples --check --diff --color --pyink-indentation=2 --line-length=125