Use np.full to create a masking array

michaelosthege · michaelosthege · commit 96bd3c37de57 · 2024-01-17T23:52:19.000+01:00
Part of #102
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:

Original file line number	Diff line number	Diff line change
`@@ -238,7 +238,7 @@ def to_inferencedata(self, , equalize_chain_lengths: bool = True, *kwargs) ->`
`238`	`238`	`_log.warning(`
`239`	`239`	`"No 'tune' stat found. Assuming all iterations are posterior draws."`
`240`	`240`	`)`
`241`		`- tune = numpy.repeat((chain_lengths[chain.cid],), False)`
	`241`	`+ tune = numpy.full((chain_lengths[chain.cid],), False)`
`242`	`242`
`243`	`243`	`# Split all variables draws into warmup/posterior`
`244`	`244`	`for var in variables:`