From 949e50380a67c934f1d10bcf241cb0c56d369289 Mon Sep 17 00:00:00 2001 From: Misha Brukman Date: Tue, 24 Jun 2025 14:49:09 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 775391353 --- clu/preprocess_spec_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/clu/preprocess_spec_test.py b/clu/preprocess_spec_test.py index 8e9f85f..2067795 100644 --- a/clu/preprocess_spec_test.py +++ b/clu/preprocess_spec_test.py @@ -149,7 +149,8 @@ def test_add_preprocess_fn(self): fn1 = preprocess_spec.PreprocessFn(ops=(op1, op2), only_jax_types=False) fn2 = preprocess_spec.PreprocessFn(ops=(op3,), only_jax_types=True) fn12 = fn1 + fn2 - self.assertEqual(fn12.ops, fn1.ops + fn2.ops) + # Note: `+` is not supported on Sequence[PreprocessOp]; need to use `list`. + self.assertSequenceEqual(fn12.ops, list(fn1.ops) + list(fn2.ops)) self.assertTrue(fn12.only_jax_types) def test_slice_preprocess_fn(self):