diff --git a/iris/algorithms/pyribs_algorithm_test.py b/iris/algorithms/pyribs_algorithm_test.py
index 1f44193..d025e16 100644
--- a/iris/algorithms/pyribs_algorithm_test.py
+++ b/iris/algorithms/pyribs_algorithm_test.py
@@ -209,7 +209,7 @@ def test_process_evaluations(self):
             params_evaluated=np.ones((13,)),
             value=1,
             obs_norm_buffer_data={  # pyrefly: ignore[bad-argument-type]
-                buffer.N: 1,
+                buffer.N: 1,  # pyrefly: ignore[bad-assignment]
                 buffer.STD: np.ones((8,)),
                 buffer.MEAN: np.ones((8,)),
                 buffer.UNNORM_VAR: np.ones((8,)),
@@ -220,7 +220,7 @@ def test_process_evaluations(self):
             params_evaluated=np.ones((13,) * 2),
             value=2,
             obs_norm_buffer_data={  # pyrefly: ignore[bad-argument-type]
-                buffer.N: 2,
+                buffer.N: 2,  # pyrefly: ignore[bad-assignment]
                 buffer.STD: np.ones((8,)) * 2,
                 buffer.MEAN: np.ones((8,)) * 2,
                 buffer.UNNORM_VAR: np.ones((8,)) * 2,
diff --git a/iris/policies/keras_cnn_policy_test.py b/iris/policies/keras_cnn_policy_test.py
index e6a4fc9..62bc788 100644
--- a/iris/policies/keras_cnn_policy_test.py
+++ b/iris/policies/keras_cnn_policy_test.py
@@ -39,15 +39,15 @@ def test_policy_act(self):
     image = np.ones((2, 2, 1))
     act = policy.act({  # pyrefly: ignore[bad-argument-type]
         'vision': image,
-        'sensor1': [-3, -3],
-        'sensor2': [-3, -3],
+        'sensor1': [-3, -3],  # pyrefly: ignore[bad-assignment]
+        'sensor2': [-3, -3],  # pyrefly: ignore[bad-assignment]
     })
     np.testing.assert_array_almost_equal(act, np.ones((5)), 1)  # pyrefly: ignore[bad-argument-type]
     policy.update_weights(new_weights=np.zeros(38))
     act = policy.act({  # pyrefly: ignore[bad-argument-type]
         'vision': image,
-        'sensor1': [-3, -3],
-        'sensor2': [-3, -3],
+        'sensor1': [-3, -3],  # pyrefly: ignore[bad-assignment]
+        'sensor2': [-3, -3],  # pyrefly: ignore[bad-assignment]
     })
     np.testing.assert_array_almost_equal(act, np.zeros((5)), 1)  # pyrefly: ignore[bad-argument-type]
 
diff --git a/iris/policies/keras_pi_policy_test.py b/iris/policies/keras_pi_policy_test.py
index f5b16d3..d532c98 100644
--- a/iris/policies/keras_pi_policy_test.py
+++ b/iris/policies/keras_pi_policy_test.py
@@ -43,16 +43,16 @@ def test_policy_act(self):
     image = np.ones((2, 2, 1))
     act = policy.act({  # pyrefly: ignore[bad-argument-type]
         'vision': image,
-        'sensor1': [-3, -3],
-        'sensor2': [-3, -3],
+        'sensor1': [-3, -3],  # pyrefly: ignore[bad-assignment]
+        'sensor2': [-3, -3],  # pyrefly: ignore[bad-assignment]
     })
     np.testing.assert_array_almost_equal(act, np.ones((5)), 1)  # pyrefly: ignore[bad-argument-type]
     policy.update_weights(np.zeros(21))
     policy.update_representation_weights(np.zeros(1001))
     act = policy.act({  # pyrefly: ignore[bad-argument-type]
         'vision': image,
-        'sensor1': [-3, -3],
-        'sensor2': [-3, -3],
+        'sensor1': [-3, -3],  # pyrefly: ignore[bad-assignment]
+        'sensor2': [-3, -3],  # pyrefly: ignore[bad-assignment]
     })
     np.testing.assert_array_almost_equal(act, np.zeros((5)), 1)  # pyrefly: ignore[bad-argument-type]
 
diff --git a/iris/workers/worker_util_test.py b/iris/workers/worker_util_test.py
index 5fd657e..5225adb 100644
--- a/iris/workers/worker_util_test.py
+++ b/iris/workers/worker_util_test.py
@@ -25,7 +25,7 @@ def test_merge(self):
         params_evaluated=np.zeros(6),
         value=np.float64(5.0),
         obs_norm_buffer_data={  # pyrefly: ignore[bad-argument-type]
-            'n': 5,
+            'n': 5,  # pyrefly: ignore[bad-assignment]
             'mean': np.zeros(7),
             'unnorm_var': np.ones(7),
         },
@@ -36,7 +36,7 @@ def test_merge(self):
         params_evaluated=np.zeros(6),
         value=np.float64(10.0),
         obs_norm_buffer_data={  # pyrefly: ignore[bad-argument-type]
-            'n': 10,
+            'n': 10,  # pyrefly: ignore[bad-assignment]
             'mean': np.ones(7),
             'unnorm_var': 2 * np.ones(7),
         },