def extract_data_from_pymc3()

in pplbench/ppls/pymc3/robust_regression.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

    def extract_data_from_pymc3(self, samples: MultiTrace) -> xr.Dataset:
        return xr.Dataset(
            {
                "alpha": (["draw"], samples["alpha"]),
                "beta": (["draw", "feature"], samples["beta"]),
                "nu": (["draw"], samples["nu"]),
                "sigma": (["draw"], samples["sigma"]),
            },
            coords={
                "draw": np.arange(samples["beta"].shape[0]),
                "feature": np.arange(samples["beta"].shape[1]),
            },
        )