def METHOD_NAME(self, draws, chains):
    class Data:
        # fake 8-school output
        shapes: Mapping[str, list] = {"mu": [], "tau": [], "eta": [8], "theta": [8]}
        obj = {key: np.random.randn(chains, draws, *shape) for key, shape in shapes.items()}
    return Data