Sandbox::https://github.com/AI-Hypercomputer/maxdiffusion/blob/main/code

Repository navigation

executable file

40 lines (33 loc) · 1.33 KB

# Copyright 2023 Google LLC
# 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.
# Clean up Python codes using Pylint & Pyink
# Googlers: please run `sudo apt install pipx; pipx install pylint --force; pipx install pyink==23.10.0` in advance
set -e # Exit immediately if any command fails
FOLDERS_TO_FORMAT=("src/maxdiffusion" "end_to_end/tpu")
LINE_LENGTH=$(grep -E "^max-line-length=" pylintrc | cut -d '=' -f 2)
# Check for --check flag
CHECK_ONLY_PYINK_FLAGS=""
if [[ "$1" == "--check" ]]; then
  CHECK_ONLY_PYINK_FLAGS="--check --diff --color"
for folder in "${FOLDERS_TO_FORMAT[@]}"
  pyink "$folder" ${CHECK_ONLY_PYINK_FLAGS} --pyink-indentation=2 --line-length=${LINE_LENGTH}
for folder in "${FOLDERS_TO_FORMAT[@]}"
  # pylint doesn't change files, only reports errors.
  pylint "./$folder"
echo "Successfully clean up all codes."

FilesExpand file tree

code_style.sh

Latest commit

History

code_style.sh

File metadata and controls