another issue bruh

zoomlogo · Feb 28, 2023 · 7c959fe · 7c959fe
1 parent ec23cf9
commit 7c959fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flax/builtins.py b/flax/builtins.py
@@ -897,8 +897,8 @@
             attrdict(
                 arity=1,
                 call=lambda x: variadic_link(
-                    links[0].call,
-                    (transpose(variadic_link(links[0].call, (x,), force_monad=True)),),
+                    links[0],
+                    (transpose(variadic_link(links[0], (x,), force_monad=True)),),
                     force_monad=True,
                 ),
             )