fix: _resolve_features check for future incompatabilities

Signed-off-by: Mehant Kammakomati <[email protected]>

tuning/utils/preprocessing_utils.py

@@ -48,7 +48,12 @@ def is_pretokenized_dataset(data: Union[str, Dataset, IterableDataset]):
             raise DatasetGenerationError("failed to load the provided dataset") from e
     if "column_names" not in data or data.column_names is None:
         if isinstance(data, IterableDataset):
-            data = data._resolve_features()
+            if hasattr(data, "_resolve_features"):
+                data = data._resolve_features()
+            else:
+                raise ValueError(
+                    "_resolve_features API is not available to fetch column names"
+                )
         else:
             raise ValueError(
                 f"not possible to fetch column names for the loaded dataset of type {type(data)}"