Skip to content

Commit 48f2e6e

Browse files
authored
Merge pull request #18 from pymc-devs/stats_types
change type to object
2 parents 697f8dc + 63568ff commit 48f2e6e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: pymc_experimental/bart/pgbart.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ class PGBART(ArrayStepShared):
5353
name = "pgbart"
5454
default_blocked = False
5555
generates_stats = True
56-
stats_dtypes = [{"variable_inclusion": np.ndarray, "bart_trees": np.ndarray}]
56+
stats_dtypes = [{"variable_inclusion": object, "bart_trees": object}]
5757

5858
def __init__(self, vars=None, num_particles=40, max_stages=100, batch="auto", model=None):
5959
model = modelcontext(model)

0 commit comments

Comments
 (0)