add more MoE tests #4279

	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

Provide feedback