Skip to content

Commit 949e503

Browse files
mbrukmancopybara-github
authored andcommitted
No public description
PiperOrigin-RevId: 775391353
1 parent 35987c1 commit 949e503

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

clu/preprocess_spec_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,8 @@ def test_add_preprocess_fn(self):
149149
fn1 = preprocess_spec.PreprocessFn(ops=(op1, op2), only_jax_types=False)
150150
fn2 = preprocess_spec.PreprocessFn(ops=(op3,), only_jax_types=True)
151151
fn12 = fn1 + fn2
152-
self.assertEqual(fn12.ops, fn1.ops + fn2.ops)
152+
# Note: `+` is not supported on Sequence[PreprocessOp]; need to use `list`.
153+
self.assertSequenceEqual(fn12.ops, list(fn1.ops) + list(fn2.ops))
153154
self.assertTrue(fn12.only_jax_types)
154155

155156
def test_slice_preprocess_fn(self):

0 commit comments

Comments
 (0)