Update test_linear.py

ASEM000 · Apr 11, 2024 · 15c3e88 · 15c3e88
1 parent 3cbd820
commit 15c3e88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_linear.py b/tests/test_linear.py
@@ -221,7 +221,7 @@ class PlusOne(NamedTuple):
     def _(
         input: jax.Array,
         weight: PlusOne,
-        bias: jax.Array | None,
+        bias,
         in_axis: Sequence[int],
         out_axis: Sequence[int],
     ):