Skip to content

Commit

Permalink
fix: Avoid numpy compatibility warning
Browse files Browse the repository at this point in the history
  • Loading branch information
aseyboldt committed Jan 14, 2025
1 parent 50bb956 commit b5a7052
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/nutpie/compile_pymc.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ def seeded_array_fn(seed: SeedType = None):
for name, shape in zip(names, shapes, strict=True):
initial_value = initial_value_dict[name]
n = int(np.prod(initial_value.shape))
if initial_value.shape != shape:
if tuple(initial_value.shape) != tuple(shape):
raise ValueError(
f"Size of initial value for {name} is {initial_value.shape}, "
f"expected {shape}"
Expand Down

0 comments on commit b5a7052

Please sign in to comment.