Skip to content

Commit 1716c0e

Browse files
author
Juan Orduz
committed
fix type
1 parent 751ec68 commit 1716c0e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pymc_bart/pgbart.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -739,7 +739,10 @@ def are_whole_number(array: npt.NDArray) -> np.bool_:
739739

740740

741741
def logp(
742-
point, out_vars: list[pm.Distribution], vars: list[pm.Distribution], shared: list[pt.Tensor]
742+
point,
743+
out_vars: list[pm.Distribution],
744+
vars: list[pm.Distribution],
745+
shared: list[pt.TensorVariable],
743746
):
744747
"""Compile PyTensor function of the model and the input and output variables.
745748

0 commit comments

Comments
 (0)