From efad567ebd3cc2df72d45ffb3cdf7dd5a8e52cc9 Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Wed, 25 Jun 2025 12:57:48 -0400
Subject: [PATCH] ruff check whole `examples/mlperf/` (#10979)

---
 .github/workflows/test.yml       | 2 +-
 examples/mlperf/helpers.py       | 2 +-
 examples/mlperf/initializers.py  | 2 +-
 examples/mlperf/lr_schedulers.py | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 417734c22a..14ccbba2e8 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -336,7 +336,7 @@ jobs:
       run: |
         pip3 install --upgrade --force-reinstall ruff==0.11.0
         python3 -m ruff check .
-        python3 -m ruff check examples/mlperf/model_train.py --ignore E501
+        python3 -m ruff check examples/mlperf/ --ignore E501
     - name: Lint tinygrad with pylint
       run: python -m pylint tinygrad/
     - name: Run mypy
diff --git a/examples/mlperf/helpers.py b/examples/mlperf/helpers.py
index f660f1481b..4ced4683b2 100644
--- a/examples/mlperf/helpers.py
+++ b/examples/mlperf/helpers.py
@@ -212,7 +212,7 @@ def get_mlperf_bert_model():
   from examples.mlperf.initializers import LinearBert, EmbeddingBert, LayerNormBert
 
   bert.Linear = LinearBert
-  bert.Embedding = EmbeddingBert 
+  bert.Embedding = EmbeddingBert
   bert.LayerNorm = LayerNormBert
 
   from extra.models.bert import BertForPretraining
diff --git a/examples/mlperf/initializers.py b/examples/mlperf/initializers.py
index 41554cba46..d84e64bab9 100644
--- a/examples/mlperf/initializers.py
+++ b/examples/mlperf/initializers.py
@@ -39,7 +39,7 @@ class LinearBert(nn.Linear):
   def __init__(self, in_features, out_features, bias=True, std=0.02):
     self.weight = std * rand_truncn(out_features, in_features, dtype=dtypes.float32)
     self.bias = Tensor.zeros(out_features, dtype=dtypes.float32) if bias else None
-  
+
   def __call__(self, x:Tensor):
     return x.cast(dtypes.default_float).linear(self.weight.cast(dtypes.default_float).transpose(), self.bias.cast(dtypes.default_float) if self.bias is not None else None)
 
diff --git a/examples/mlperf/lr_schedulers.py b/examples/mlperf/lr_schedulers.py
index 600924f526..f21b06a5ba 100644
--- a/examples/mlperf/lr_schedulers.py
+++ b/examples/mlperf/lr_schedulers.py
@@ -1,4 +1,4 @@
-from tinygrad import Tensor, dtypes
+from tinygrad import dtypes
 from tinygrad.nn.optim import Optimizer
 
 from extra.lr_scheduler import LR_Scheduler