From d33c3cfc765a18950194205a1ddb92a0981a355e Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Tue, 25 Aug 2026 04:51:34 -0700
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 970473925
---
 clrs/examples/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/clrs/examples/run.py b/clrs/examples/run.py
index 38fea7aa..776bf463 100644
--- a/clrs/examples/run.py
+++ b/clrs/examples/run.py
@@ -435,7 +435,7 @@ def main(unused_argv):
       rng=rng,
       train_lengths=train_lengths,
       algorithms=FLAGS.algorithms,
-      val_lengths=[np.amax(train_lengths)],
+      val_lengths=[np.amax(train_lengths)],  # pyrefly: ignore[bad-argument-type]
       test_lengths=[-1],
       train_batch_size=FLAGS.batch_size,
   )