Skip to content

Commit 345e3ce

Browse files
Default to using McBackend
1 parent 4b1f5d3 commit 345e3ce

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pymc/backends/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@
7575

7676
HAS_MCB = False
7777
try:
78-
from mcbackend import Backend, Run
78+
from mcbackend import Backend, NumPyBackend, Run
7979

8080
from pymc.backends.mcbackend import init_chain_adapters
8181

@@ -123,6 +123,8 @@ def init_traces(
123123
model: Model,
124124
) -> Tuple[Optional[RunType], Sequence[IBaseTrace]]:
125125
"""Initializes a trace recorder for each chain."""
126+
if HAS_MCB and backend is None:
127+
backend = NumPyBackend(preallocate=expected_length)
126128
if HAS_MCB and isinstance(backend, Backend):
127129
return init_chain_adapters(
128130
backend=backend,

0 commit comments

Comments
 (0)