tensor(broadcastable=...) kwarg is depreciated (#710)

Author: jessegrabowski

preliz/internal/distribution_helper.py

@@ -125,7 +125,8 @@ def inner_func(*args, signature_to_function=signature_to_function):
         except KeyError:
             pass
         symbolic_args = [
-            tensor(broadcastable=bcast_pattern, dtype=dtype) for (bcast_pattern, dtype) in signature
+            tensor(shape=tuple(1 if b else None for b in bcast_pattern), dtype=dtype)
+            for (bcast_pattern, dtype) in signature
         ]
         symbolic_out = func(*symbolic_args)
         signature_to_function[signature] = compiled_func = function(
@@ -169,7 +170,7 @@ def inner_func(*args, size, rng, signature_to_function=signature_to_function):
                 pass
 
             symbolic_args = [
-                tensor(broadcastable=bcast_pattern, dtype=dtype)
+                tensor(shape=tuple(1 if b else None for b in bcast_pattern), dtype=dtype)
                 for (bcast_pattern, dtype) in signature[:-1]
             ]
             symbolic_size = (