def METHOD_NAME(
    continuous_search_space: ContinuousSearchSpace, dimensionality: int, num_points: int
):
    samples = continuous_search_space.sample(num_points=num_points, key=jr.PRNGKey(42))
    assert samples.shape[0] == num_points
    assert samples.shape[1] == dimensionality