fix type

pymc-devs · Dec 26, 2024 · 1716c0e · 1716c0e
1 parent 751ec68
commit 1716c0e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pymc_bart/pgbart.py b/pymc_bart/pgbart.py
@@ -739,7 +739,10 @@ def are_whole_number(array: npt.NDArray) -> np.bool_:
 
 
 def logp(
-    point, out_vars: list[pm.Distribution], vars: list[pm.Distribution], shared: list[pt.Tensor]
+    point,
+    out_vars: list[pm.Distribution],
+    vars: list[pm.Distribution],
+    shared: list[pt.TensorVariable],
 ):
     """Compile PyTensor function of the model and the input and output variables.