Skip to content

Commit b5a7052

Browse files
committed
fix: Avoid numpy compatibility warning
1 parent 50bb956 commit b5a7052

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/nutpie/compile_pymc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def seeded_array_fn(seed: SeedType = None):
6969
for name, shape in zip(names, shapes, strict=True):
7070
initial_value = initial_value_dict[name]
7171
n = int(np.prod(initial_value.shape))
72-
if initial_value.shape != shape:
72+
if tuple(initial_value.shape) != tuple(shape):
7373
raise ValueError(
7474
f"Size of initial value for {name} is {initial_value.shape}, "
7575
f"expected {shape}"

0 commit comments

Comments
 (0)