From 4cd134acd2d50b06b4652b7fa9371e10486e9fa6 Mon Sep 17 00:00:00 2001
From: Jake VanderPlas <vanderplas@google.com>
Date: Thu, 6 Aug 2026 08:20:03 -0700
Subject: [PATCH] add pyrefly suppressions for jax.jit annotation change

PiperOrigin-RevId: 960313712
---
 examples/transformer/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/transformer/train.py b/examples/transformer/train.py
index 41e2786b1..b876d1ae1 100644
--- a/examples/transformer/train.py
+++ b/examples/transformer/train.py
@@ -177,7 +177,7 @@ def main(_):
     if step % LOG_EVERY == 0:
       steps_per_sec = LOG_EVERY / (time.time() - prev_time)
       prev_time = time.time()
-      metrics |= {'steps_per_sec': steps_per_sec}
+      metrics |= {'steps_per_sec': steps_per_sec}  # pyrefly: ignore[unsupported-operation]
       logging.info({k: float(v) for k, v in metrics.items()})