Remove extra return

Ian Schweer · Ian Schweer · commit fd2f192e619d · 2024-11-28T17:05:15.000-08:00
diff --git a/pytensor/link/pytorch/dispatch/elemwise.py b/pytensor/link/pytorch/dispatch/elemwise.py
@@ -46,7 +46,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
 
