import thermostat

data = thermostat.load("imdb-bert-lig", cache_dir="~/datasets")