better for mypy

williambdean · williambdean · commit 5884a39de74b · 2025-04-22T14:03:29.000-04:00
diff --git a/pymc/testing.py b/pymc/testing.py
@@ -1028,9 +1028,9 @@ def mock_pymc_sample():
         coords={"chain": np.arange(n_chains)},
     )
     idata.add_groups(
-        posterior=(idata.prior.mean("chain") * expanded_chains).transpose("chain", "draw", ...),
+        posterior=(idata["prior"].mean("chain") * expanded_chains).transpose("chain", "draw", ...),
     )
-    del idata.prior
+    del idata["prior"]
     if "prior_predictive" in idata:
-        del idata.prior_predictive
+        del idata["prior_predictive"]
     return idata

Original file line number	Diff line number	Diff line change
`@@ -1028,9 +1028,9 @@ def mock_pymc_sample():`
`1028`	`1028`	`coords={"chain": np.arange(n_chains)},`
`1029`	`1029`	`)`
`1030`	`1030`	`idata.add_groups(`
`1031`		`- posterior=(idata.prior.mean("chain") * expanded_chains).transpose("chain", "draw", ...),`
	`1031`	`+ posterior=(idata["prior"].mean("chain") * expanded_chains).transpose("chain", "draw", ...),`
`1032`	`1032`	`)`
`1033`		`- del idata.prior`
	`1033`	`+ del idata["prior"]`
`1034`	`1034`	`if "prior_predictive" in idata:`
`1035`		`- del idata.prior_predictive`
	`1035`	`+ del idata["prior_predictive"]`
`1036`	`1036`	`return idata`