split -> subset

app.py

@@ -18,16 +18,16 @@ from pymatgen.core import Structure
 HF_TOKEN = os.environ.get("HF_TOKEN")
 top_k = 500
 
-splits = ["compatible_pbe", "compatible_pbesol", "compatible_scan", "non_compatible"]
+subsets = ["compatible_pbe", "compatible_pbesol", "compatible_scan", "non_compatible"]
 
 # Load only the train split of the dataset
 
 datasets = []
-for split in splits:
+for subset in subsets:
     dataset = load_dataset(
         "LeMaterial/leMat-Bulk",
+        subset,
         token=HF_TOKEN,
-        split=split,
         columns=[
             "lattice_vectors",
             "species_at_sites",