Lint

Fix chief-0 timeout in tf_keras premade models distributed test by specifying steps_per_epoch when training on Dataset. #1459

	name: Lint

	on:
	push:
	branches: [ master ]
	pull_request:
	workflow_dispatch:

	permissions:
	contents: read # to fetch code (actions/checkout)

	jobs:
	lint:
	name: Check the code format
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v4
	- name: Set up Python 3.10
	uses: actions/setup-python@v5
	with:
	python-version: '3.10'
	- name: Get pip cache dir
	id: pip-cache
	run: \|
	python -m pip install --upgrade pip setuptools
	echo "::set-output name=dir::$(pip cache dir)"
	- name: pip cache
	uses: actions/cache@v3
	with:
	path: ${{ steps.pip-cache.outputs.dir }}
	key: ${{ runner.os }}-pip-${{ hashFiles('requirements.txt') }}
	restore-keys: \|
	${{ runner.os }}-pip-
	- name: Install dependencies
	run: \|
	pip install black==22.3.0 isort==5.10.1 flake8==4.0.1
	- name: Lint the code
	run: sh shell/lint.sh

Provide feedback