Added exception handling for dataset loading

Signed-off-by: Ann Kuruvilla <akuruvil@qti.qualcomm.com>

Author: Ann Kuruvilla

QEfficient/finetune/experimental/core/dataset.py

@@ -140,7 +140,13 @@ def _initialize_dataset(self):
             if self.config_name is not None:
                 load_kwargs["name"] = self.config_name
 
-            db = load_dataset_builder(self.dataset_name, **load_kwargs)
+            try:
+                db = load_dataset_builder(self.dataset_name, **load_kwargs)
+            except Exception as e:
+                raise RuntimeError(
+                f"Failed to load dataset builder for '{self.dataset_name}': {e}. "
+                "Please check the dataset name and your network connection."
+            )
             available_splits = []
             if db.info.splits is not None:
                 available_splits = list(db.info.splits.keys())
@@ -151,7 +157,13 @@ def _initialize_dataset(self):
             if self.split not in available_splits:
                 load_split = "train"
             # FIXME: Add streaming support for larger datasets.
-            self.dataset = load_dataset(self.dataset_name, split=load_split, **load_kwargs)
+            try:
+                self.dataset = load_dataset(self.dataset_name, split=load_split, **load_kwargs)
+            except Exception as e:
+                raise RuntimeError(
+                f"Failed to load dataset '{self.dataset_name}' with split '{load_split}': {e}. "
+                "Please verify the dataset exists and is accessible."
+            )
             self.dataset = self.dataset.shuffle(seed=self.seed)
             if self.dataset_disc_style:
                 available_styles = set(self.dataset["category"])