Skip to content

Commit 1b26499

Browse files
committed
fix failing doctest
1 parent b0eabff commit 1b26499

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

causalpy/pymc_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ class PyMCModel(pm.Model):
3737
>>> import causalpy as cp
3838
>>> import numpy as np
3939
>>> import pymc as pm
40-
>>> from causalpy.pymc_models import ModelBuilder
41-
>>> class MyToyModel(ModelBuilder):
40+
>>> from causalpy.pymc_models import PyMCModel
41+
>>> class MyToyModel(PyMCModel):
4242
... def build_model(self, X, y, coords):
4343
... with self:
4444
... X_ = pm.Data(name="X", value=X)

0 commit comments

Comments
 (0)