def METHOD_NAME(wait, client):
    df = dask.datasets.timeseries(
        start="2000",
        end="2001",
        dtypes={"value": float, "name": str, "id": int},
        freq="2H",
        partition_freq="1M",
        seed=1,
    )
    df = df.persist()
    wait(df)
    df2 = df.set_index("name", shuffle="tasks")
    df2 = df2.persist()
    assert len(df2)