Skip to content

Commit ba5f3a1

Browse files
committed
Improve error message of get_support_shape
1 parent c120f7e commit ba5f3a1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pymc/distributions/shape_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -734,7 +734,7 @@ def get_support_shape(
734734
elif support_shape is not None:
735735
inferred_support_shape = at.stack(
736736
[
737-
Assert(msg="support_shape does not match last shape dimension")(
737+
Assert(msg="support_shape does not match respective shape dimension")(
738738
inferred, at.eq(inferred, explicit)
739739
)
740740
for inferred, explicit in zip(inferred_support_shape, support_shape)

pymc/tests/distributions/test_timeseries.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def test_missing_steps(self):
123123

124124
def test_inconsistent_steps_and_shape(self):
125125
with pytest.raises(
126-
AssertionError, match="support_shape does not match last shape dimension"
126+
AssertionError, match="support_shape does not match respective shape dimension"
127127
):
128128
x = GaussianRandomWalk.dist(steps=12, shape=45, init_dist=Normal.dist(0, 100))
129129

0 commit comments

Comments
 (0)