This version doesn't have a dependency on dask and avoids the dask
version conflict with merlin-core