Use np.full to create a masking array

Part of #102
pymc-devs · Jan 17, 2024 · 96bd3c3 · 96bd3c3
1 parent 9fa6fb6
commit 96bd3c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mcbackend/core.py b/mcbackend/core.py
@@ -238,7 +238,7 @@ def to_inferencedata(self, *, equalize_chain_lengths: bool = True, **kwargs) ->
                     _log.warning(
                         "No 'tune' stat found. Assuming all iterations are posterior draws."
                     )
-                tune = numpy.repeat((chain_lengths[chain.cid],), False)
+                tune = numpy.full((chain_lengths[chain.cid],), False)
 
             # Split all variables draws into warmup/posterior
             for var in variables: