Skip to content

Commit dc1ef8e

Browse files
committed
bugfix: ensure type before jnp.tile
1 parent 40ebf6d commit dc1ef8e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/pyhf/tensor/jax_backend.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,8 @@ def tile(self, tensor_in, repeats):
157157
Returns:
158158
JAX ndarray: The tensor with repeated axes
159159
"""
160+
if not isinstance(tensor_in, jnp.ndarray):
161+
tensor_in = jnp.array(tensor_in)
160162
return jnp.tile(tensor_in, repeats)
161163

162164
def conditional(self, predicate, true_callable, false_callable):

0 commit comments

Comments
 (0)