fix

prepare.py

@@ -13,6 +13,8 @@ if __name__ == "__main__":
         # Load dataset using datasets.load_dataset()
         dataset = datasets.load_dataset("renumics/cifar100-enriched", split="train")
         print("Dataset loaded using datasets.load_dataset().")
+        
+        df = dataset.to_pandas()  
 
         # Save dataset to cache
         df.to_parquet(cache_file)

run.py

@@ -14,6 +14,8 @@ if __name__ == "__main__":
         # Load dataset using datasets.load_dataset()
         dataset = datasets.load_dataset("renumics/cifar100-enriched", split="train")
         print("Dataset loaded using datasets.load_dataset().")
+        
+        df = dataset.to_pandas()  
 
         # Save dataset to cache
         df.to_parquet(cache_file)