From 0bbdfb38f4046b2c7026e69541c68537ccf6abf2 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Tue, 25 Aug 2026 04:51:34 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 970473925 --- iris/algorithms/multi_agent_ars_algorithm_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/iris/algorithms/multi_agent_ars_algorithm_test.py b/iris/algorithms/multi_agent_ars_algorithm_test.py index 1cc662d..25c5b28 100644 --- a/iris/algorithms/multi_agent_ars_algorithm_test.py +++ b/iris/algorithms/multi_agent_ars_algorithm_test.py @@ -523,15 +523,15 @@ def test_split_checkpoint(self): if expected_states[i]['obs_norm_state'] is not None: self.assertAllClose( agent_state['obs_norm_state']['mean'], - expected_states[i]['obs_norm_state']['mean'], + expected_states[i]['obs_norm_state']['mean'], # pyrefly: ignore[bad-index] ) self.assertAllClose( agent_state['obs_norm_state']['std'], - expected_states[i]['obs_norm_state']['std'], + expected_states[i]['obs_norm_state']['std'], # pyrefly: ignore[bad-index] ) self.assertAllClose( agent_state['obs_norm_state']['n'], - expected_states[i]['obs_norm_state']['n'], + expected_states[i]['obs_norm_state']['n'], # pyrefly: ignore[bad-index] )