From eec180ddf65c4dba107c1a5a091a85628007db13 Mon Sep 17 00:00:00 2001 From: Oleh Prypin Date: Tue, 20 Oct 2026 03:30:00 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 1002975158 --- learned_optimization/baselines/normalizers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/learned_optimization/baselines/normalizers.py b/learned_optimization/baselines/normalizers.py index 866f868c..b6c75e07 100644 --- a/learned_optimization/baselines/normalizers.py +++ b/learned_optimization/baselines/normalizers.py @@ -53,10 +53,10 @@ def ema(data: chex.Array, alpha: float, ignore_nan=False): m_alpha = alpha # TODO(lmetz) profile if this is needed / saves much time. if ignore_nan: - for i, a in enumerate((1 - alpha) * data[1:]): + for i, a in enumerate((1 - alpha) * data[1:]): # pyrefly: ignore[not-iterable, unsupported-operation] x[i + 1] = x[i] if onp.isnan(a) else x[i] * m_alpha + a else: - for i, a in enumerate((1 - alpha) * data[1:]): + for i, a in enumerate((1 - alpha) * data[1:]): # pyrefly: ignore[not-iterable, unsupported-operation] x[i + 1] = x[i] * m_alpha + a return x