Add linter script internally and fix lint issues

PiperOrigin-RevId: 953429485

Author: SurbhiJainUSC

src/maxtext/checkpoint_conversion/utils/param_mapping.py

@@ -29,7 +29,7 @@
     **Value: corresponding Hugging Face parameters, with following forms:**
     First, the base element mapped to can be either:
     - `atomic_hf_key`: A single string representing one Hugging Face parameter.
-    - `composite_hf_key`: A tuple of strings representing multiple Hugging Face parameters that combine 
+    - `composite_hf_key`: A tuple of strings representing multiple Hugging Face parameters that combine
     into a single MaxText parameter (e.g., Qwen's qkv and z).
 
     These base elements (strings or tuples) are then structured as:

src/maxtext/examples/sft_qwen3_demo.ipynb

@@ -21,7 +21,7 @@
     "\n",
     "This notebook performs SFT training and evaluation workflow on [OpenAI's GSM8K dataset](https://huggingface.co/datasets/openai/gsm8k).\n",
     "The primary goal is to demonstrate the end-to-end process of:\n",
-    "1. Pre-SFT Evaluation: Calcuating baseline accuracy for the model before training.\n",
+    "1. Pre-SFT Evaluation: Calculating baseline accuracy for the model before training.\n",
     "2. SFT Training: Fine-tune the model using MaxText & Tunix SFT trainer.\n",
     "3. Post-SFT Evaluation: Re-running the evaluation loop after training to measure the performance gain achieved by SFT.\n",
     "\n",
@@ -363,7 +363,7 @@
     "        f\"Processing {NUM_TEST_SAMPLES} examples with a batch size of {BATCH_SIZE}. This will result in {TOTAL_BATCHES} total batches for the test run.\"\n",
     "    )\n",
     "else:\n",
-    "    print(\"Evaluation on test dataset is skipped. Plese set `RUN_EVALUATION=True` to run evaluation.\")"
+    "    print(\"Evaluation on test dataset is skipped. Please set `RUN_EVALUATION=True` to run evaluation.\")"
    ]
   },
   {
@@ -419,7 +419,7 @@
     "        ),\n",
     "    )\n",
     "else:\n",
-    "    print(\"Evaluation on test dataset is skipped. Plese set `RUN_EVALUATION=True` to run evaluation.\")"
+    "    print(\"Evaluation on test dataset is skipped. Please set `RUN_EVALUATION=True` to run evaluation.\")"
    ]
   },
   {
@@ -443,7 +443,7 @@
     "    print(\"Running Pre-SFT Evaluation...\")\n",
     "    score = evaluate_model(test_dataset, vllm_rollout, debug=False)\n",
     "else:\n",
-    "    print(\"Evaluation on test dataset is skipped. Plese set `RUN_EVALUATION=True` to run evaluation.\")"
+    "    print(\"Evaluation on test dataset is skipped. Please set `RUN_EVALUATION=True` to run evaluation.\")"
    ]
   },
   {
@@ -464,7 +464,7 @@
     "        f\"Percentage of test samples where the model's output adheres to the expected structure: {score['correct_format']}%\"\n",
     "    )\n",
     "else:\n",
-    "    print(\"Evaluation on test dataset is skipped. Plese set `RUN_EVALUATION=True` to run evaluation.\")"
+    "    print(\"Evaluation on test dataset is skipped. Please set `RUN_EVALUATION=True` to run evaluation.\")"
    ]
   },
   {
@@ -515,7 +515,7 @@
     "    vllm_rollout.update_params(state)\n",
     "    score = evaluate_model(test_dataset, vllm_rollout, debug=False)\n",
     "else:\n",
-    "    print(\"Evaluation on test dataset is skipped. Plese set `RUN_EVALUATION=True` to run evaluation.\")"
+    "    print(\"Evaluation on test dataset is skipped. Please set `RUN_EVALUATION=True` to run evaluation.\")"
    ]
   },
   {
@@ -538,7 +538,7 @@
     "        f\"Percentage of test samples where the model's output adheres to the expected structure: {score['correct_format']}%\"\n",
     "    )\n",
     "else:\n",
-    "    print(\"Evaluation on test dataset is skipped. Plese set `RUN_EVALUATION=True` to run evaluation.\")"
+    "    print(\"Evaluation on test dataset is skipped. Please set `RUN_EVALUATION=True` to run evaluation.\")"
    ]
   }
  ],

src/maxtext/experimental/agent/ckpt_conversion_agent/baselines/context/hf_shape.py

@@ -1,17 +1,17 @@
 """
- Copyright 2025 Google LLC
+Copyright 2025 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.
 """
 
 """Example code for agent, only gemma2 mappings"""

src/maxtext/experimental/agent/ckpt_conversion_agent/baselines/context/param_mapping.py

@@ -1,17 +1,17 @@
 """
- Copyright 2025 Google LLC
+Copyright 2025 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.
 """
 
 """Example code for agent, only gemma2 mappings"""

src/maxtext/experimental/agent/ckpt_conversion_agent/dsl.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 """
-A DSL agent class, to propose potential new DSL rules/ops based on the previous analysis call. 
+A DSL agent class, to propose potential new DSL rules/ops based on the previous analysis call.
 """
 import argparse
 import os

src/maxtext/experimental/agent/ckpt_conversion_agent/transformation.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 """
-A transformation agent to generate the 
+A transformation agent to generate the
 layerwise and bidirectional transformation hook functions between HF & Maxtext
 """
 import argparse

src/maxtext/inference/decode.py

@@ -33,6 +33,7 @@
 _config_lib, engine_api, _token_utils, _tokenizer_api, _token_params_ns = jetstream()
 # Placeholder: internal
 
+
 # Number of text sequences to process in a single batch.
 _NUM_STREAMS = 1
 

src/maxtext/inference/mlperf/evaluate-accuracy.py

@@ -11,7 +11,7 @@
 # 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.
-""" Evaluation script based on MLPerf requirements"""
+"""Evaluation script based on MLPerf requirements"""
 
 import argparse
 import json

src/maxtext/inference/mlperf/matmul/timing_util.py

@@ -11,7 +11,7 @@
 # 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.
-""" Timing utility functions """
+"""Timing utility functions"""
 
 import datetime
 import tempfile

src/maxtext/inference/mlperf/offline_inference.py

@@ -11,7 +11,7 @@
 # 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.
-""" Offline inference for mlperf """
+"""Offline inference for mlperf"""
 
 from collections import defaultdict
 from typing import Any, Callable