Default to using McBackend

pymc-devs · Mar 1, 2023 · 1988b65 · 1988b65
1 parent 5510d3f
commit 1988b65
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pymc/backends/__init__.py b/pymc/backends/__init__.py
@@ -75,7 +75,7 @@
 
 HAS_MCB = False
 try:
-    from mcbackend import Backend, Run
+    from mcbackend import Backend, NumPyBackend, Run
 
     from pymc.backends.mcbackend import init_chain_adapters
 
@@ -123,6 +123,8 @@ def init_traces(
     model: Model,
 ) -> Tuple[Optional[RunType], Sequence[IBaseTrace]]:
     """Initializes a trace recorder for each chain."""
+    if HAS_MCB and backend is None:
+        backend = NumPyBackend(preallocate=expected_length)
     if HAS_MCB and isinstance(backend, Backend):
         return init_chain_adapters(
             backend=backend,