def METHOD_NAME(self, rng):
    # 1D dataset
    X1 = np.linspace(-5, 5, 100)[:, None]
    # 3D dataset
    x1, x2, x3 = np.meshgrid(
        np.linspace(0, 10, 5), np.linspace(20, 30, 5), np.linspace(10, 20, 5)
    )
    X2 = np.vstack([x1.flatten(), x2.flatten(), x3.flatten()]).T
    return X1, X2