Merge pull request #971 from krishnan-chandra/fix-dataset-requirement

Loosen requirement on `datasets` package

Author: arnavsinghvi11