Remove extra return

Ian Schweer · Ian Schweer · commit 7dd9edba8429 · 2024-11-25T09:34:47.000-08:00
diff --git a/pytensor/link/pytorch/dispatch/elemwise.py b/pytensor/link/pytorch/dispatch/elemwise.py
@@ -32,7 +32,6 @@ def elemwise_fn(*inputs):
             for _ in range(broadcast_inputs[0].dim()):
                 ufunc = torch.vmap(ufunc)
             return ufunc(*broadcast_inputs)
-            return base_fn(*inputs)
 
     return elemwise_fn
 
