adding script to fix the style and adding modified/fixed files

.github/workflows/CPUTests.yaml

@@ -0,0 +1,38 @@
+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; }

.github/workflows/UnitTests.yml

@@ -27,31 +27,6 @@ on:
  - cron: '0 */2 * * *'
 
 jobs:
- cpu:
- name: "CPU test"
- runs-on: ${{ matrix.os }}
- strategy:
- matrix:
- os: [ubuntu-20.04]
- python-version: ['3.10']
- steps:
- - uses: actions/checkout@v3
- - name: setup python
- uses: actions/setup-python@v4
- with:
- python-version: ${{ matrix.python-version }}
- - name: Install Dependencies
- run: |
- pip install pytype==2024.2.27
- pip install pylint
- - name: Typecheck the code with pytype
- run: |
- pytype --jobs auto --disable import-error MaxText/
- - name: Analysing the code with pylint
- run: |
- pylint MaxText/
-
-
  # IF YOU MODIFY THIS, YOU SHOULD ALSO ADD CORRESPONDING MODICATIONS TO 'gpu' job
  tpu:
  strategy:

MaxText/__init__.py

@@ -1,15 +1,15 @@
 """
- Copyright 2023 Google LLC
+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
+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
 
-  https://www.apache.org/licenses/LICENSE-2.0
+ https://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.
- """
+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.
+"""