nextrec 0.2.4__py3-none-any.whl → 0.2.6__py3-none-any.whl

nextrec/data/data_utils.py

@@ -6,32 +6,64 @@ import pandas as pd
 import pyarrow.parquet as pq
 from pathlib import Path
 
+
+def _stack_section(batch: list[dict], section: str):
+    """Stack one section of the batch (features/labels/ids)."""
+    entries = [item.get(section) for item in batch if item.get(section) is not None]
+    if not entries:
+        return None
+    merged: dict = {}
+    for name in entries[0]:
+        tensors = [item[section][name] for item in batch if item.get(section) is not None and name in item[section]]
+        merged[name] = torch.stack(tensors, dim=0)
+    return merged
+
+
 def collate_fn(batch):
-    """Collate a list of tensor tuples from ``FileDataset`` into batched tensors."""
+    """
+    Collate a list of sample dicts into the unified batch format:
+    {
+        "features": {name: Tensor(B, ...)},
+        "labels": {target: Tensor(B, ...)} or None,
+        "ids": {id_name: Tensor(B, ...)} or None,
+    }
+    """
     if not batch:
-        return tuple()
-
-    num_tensors = len(batch[0])
+        return {"features": {}, "labels": None, "ids": None}
+
+    first = batch[0]
+    if isinstance(first, dict) and "features" in first:
+        # Streaming dataset yields already-batched chunks; avoid adding an extra dim.
+        if first.get("_already_batched") and len(batch) == 1:
+            return {
+                "features": first.get("features", {}),
+                "labels": first.get("labels"),
+                "ids": first.get("ids"),
+            }
+        return {
+            "features": _stack_section(batch, "features") or {},
+            "labels": _stack_section(batch, "labels"),
+            "ids": _stack_section(batch, "ids"),
+        }
+
+    # Fallback: stack tuples/lists of tensors
+    num_tensors = len(first)
     result = []
-    
     for i in range(num_tensors):
         tensor_list = [item[i] for item in batch]
-        first = tensor_list[0]
-
-        if isinstance(first, torch.Tensor):
+        first_item = tensor_list[0]
+        if isinstance(first_item, torch.Tensor):
             stacked = torch.cat(tensor_list, dim=0)
-        elif isinstance(first, np.ndarray):
+        elif isinstance(first_item, np.ndarray):
             stacked = np.concatenate(tensor_list, axis=0)
-        elif isinstance(first, list):
+        elif isinstance(first_item, list):
             combined = []
             for entry in tensor_list:
                 combined.extend(entry)
             stacked = combined
         else:
             stacked = tensor_list
-
         result.append(stacked)
-    
     return tuple(result)
 
 

nextrec/data/dataloader.py

@@ -2,6 +2,7 @@
 Dataloader definitions
 
 Date: create on 27/10/2025
+Update: 25/11/2025
 Author: Yang Zhou,zyaztec@gmail.com
 """
 import os
@@ -15,7 +16,7 @@ import pyarrow.parquet as pq
 from pathlib import Path
 from typing import Iterator, Literal, Union, Optional
 
-from torch.utils.data import DataLoader, TensorDataset, IterableDataset
+from torch.utils.data import DataLoader, Dataset, IterableDataset
 from nextrec.data.preprocessor import DataProcessor
 from nextrec.basic.features import DenseFeature, SparseFeature, SequenceFeature, FeatureSpecMixin
 
@@ -27,27 +28,33 @@ from nextrec.data import (
     read_table,
 )
 
+class TensorDictDataset(Dataset):
+    """Dataset returning sample-level dicts matching the unified batch schema."""
+    def __init__(self, tensors: dict):
+        self.features = tensors.get("features", {})
+        self.labels = tensors.get("labels")
+        self.ids = tensors.get("ids")
+        if not self.features:
+            raise ValueError("Dataset requires at least one feature tensor.")
+        lengths = [tensor.shape[0] for tensor in self.features.values()]
+        if not lengths:
+            raise ValueError("Feature tensors are empty.")
+        self.length = lengths[0]
+        for length in lengths[1:]:
+            if length != self.length:
+                raise ValueError("All feature tensors must have the same length.")
+
+    def __len__(self) -> int:
+        return self.length
+
+    def __getitem__(self, idx: int) -> dict:
+        sample_features = {name: tensor[idx] for name, tensor in self.features.items()}
+        sample_labels = {name: tensor[idx] for name, tensor in self.labels.items()} if self.labels else None
+        sample_ids = {name: tensor[idx] for name, tensor in self.ids.items()} if self.ids else None
+        return {"features": sample_features, "labels": sample_labels, "ids": sample_ids}
 
-class FileDataset(FeatureSpecMixin, IterableDataset):
-    """
-    Iterable dataset that streams CSV/Parquet files in chunks and yields tensor tuples.
-
-    :param file_paths: Absolute or relative paths to CSV/Parquet files.
-    :param dense_features: Dense feature definitions (float tensors).
-    :param sparse_features: Sparse/categorical feature definitions (int tensors).
-    :param sequence_features: Sequence feature definitions (padded int tensors).
-    :param target_columns: Label/target column names.
-    :param id_columns: Optional ID columns appended after targets.
-    :param chunk_size: Number of rows to read per chunk.
-    :param file_type: ``\"csv\"`` or ``\"parquet\"``.
-    :param processor: Optional fitted :class:`~nextrec.data.preprocessor.DataProcessor` for online transform.
 
-    Yields
-    ------
-    tuple
-        Tensors ordered as ``dense + sparse + sequence + targets (+ ids)``. Shape respects chunk size.
-    """
-    
+class FileDataset(FeatureSpecMixin, IterableDataset):
     def __init__(self, 
                  file_paths: list[str],                      # file paths to read, containing CSV or Parquet files
                  dense_features: list[DenseFeature],         # dense feature definitions
@@ -58,95 +65,48 @@ class FileDataset(FeatureSpecMixin, IterableDataset):
                  chunk_size: int = 10000,
                  file_type: str = 'csv',
                  processor: DataProcessor | None = None): # optional DataProcessor for transformation
-        """
-        Initialize a streaming dataset backed by on-disk files.
-        """
-
         self.file_paths = file_paths
         self.chunk_size = chunk_size
         self.file_type = file_type
         self.processor = processor
-
-        self._set_feature_config(dense_features, sparse_features, sequence_features)
-        self._set_target_config(target_columns, id_columns or [])
+        self._set_feature_config(dense_features, sparse_features, sequence_features, target_columns, id_columns)
         self.current_file_index = 0
         self.total_files = len(file_paths)
     
-    def __iter__(self) -> Iterator[tuple]:
-        """
-        Iterate over files and stream tensor tuples chunk by chunk.
-
-        Files are processed sequentially; each chunk is transformed (optionally via
-        ``processor``) and converted to tensors before being yielded to PyTorch ``DataLoader``.
-        """
+    def __iter__(self):
         self.current_file_index = 0
         self._file_pbar = None
-        
-        # Create progress bar for file processing when multiple files
         if self.total_files > 1:
-            self._file_pbar = tqdm.tqdm(
-                total=self.total_files, 
-                desc="Files", 
-                unit="file",
-                position=0,
-                leave=True,
-                bar_format='{desc}: {percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}]'
-            )
-        
+            self._file_pbar = tqdm.tqdm(total=self.total_files, desc="Files", unit="file", position=0, leave=True, bar_format='{desc}: {percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}]')
         for file_path in self.file_paths:
             self.current_file_index += 1
-            
             if self._file_pbar is not None:
                 self._file_pbar.update(1)
             elif self.total_files == 1:
                 file_name = os.path.basename(file_path)
-                logging.info(colorize(f"Processing file: {file_name}", color="cyan"))
-            
+                logging.info(f"Processing file: {file_name}")
             if self.file_type == 'csv':
                 yield from self._read_csv_chunks(file_path)
             elif self.file_type == 'parquet':
                 yield from self._read_parquet_chunks(file_path)
-        
         if self._file_pbar is not None:
             self._file_pbar.close()
     
-    def _read_csv_chunks(self, file_path: str) -> Iterator[tuple]:
-        """
-        Stream a CSV file chunk by chunk.
-
-        :param file_path: Path to the CSV file.
-        :yields: Tensor tuples for each chunk.
-        """
+    def _read_csv_chunks(self, file_path: str):
         chunk_iterator = pd.read_csv(file_path, chunksize=self.chunk_size)
-        
         for chunk in chunk_iterator:
             tensors = self._dataframe_to_tensors(chunk)
-            if tensors:
-                yield tensors
+            yield tensors
     
-    def _read_parquet_chunks(self, file_path: str) -> Iterator[tuple]:
-        """
-        Stream a Parquet file via ``pyarrow`` batch reading.
-
-        :param file_path: Path to the Parquet file.
-        :yields: Tensor tuples for each batch.
-        """
-
+    def _read_parquet_chunks(self, file_path: str):
         parquet_file = pq.ParquetFile(file_path)
         for batch in parquet_file.iter_batches(batch_size=self.chunk_size):
             chunk = batch.to_pandas()            
             tensors = self._dataframe_to_tensors(chunk)
-            if tensors:
-                yield tensors
+            yield tensors
             del chunk
                 
-    def _dataframe_to_tensors(self, df: pd.DataFrame) -> tuple | None:
-        """
-        Convert a DataFrame chunk into a tuple of tensors respecting feature order.
-
-        :param df: DataFrame chunk.
-        :returns: Tuple of tensors (features + targets + ids) or ``None`` if no tensors created.
-        """
+    def _dataframe_to_tensors(self, df: pd.DataFrame) -> dict | None:
         if self.processor is not None:
             if not self.processor.is_fitted:
                 raise ValueError("DataProcessor must be fitted before using in streaming mode")
@@ -154,44 +114,19 @@ class FileDataset(FeatureSpecMixin, IterableDataset):
         else:
             transformed_data = df
 
-        return build_tensors_from_data(
+        batch = build_tensors_from_data(
             data=transformed_data,
             raw_data=df,
             features=self.all_features,
             target_columns=self.target_columns,
             id_columns=self.id_columns,
-            on_missing_feature="raise",
         )
+        if batch is not None:
+            batch["_already_batched"] = True
+        return batch
 
 
 class RecDataLoader(FeatureSpecMixin):
-    """
-    Convenience wrapper for building PyTorch ``DataLoader`` objects for recommendation models.
-
-    :param dense_features: Dense feature definitions (float tensors).
-    :param sparse_features: Sparse/categorical feature definitions (int tensors).
-    :param sequence_features: Sequence feature definitions (padded int tensors).
-    :param target: Target column name(s); string or list.
-    :param id_columns: Optional ID column name(s) appended after targets.
-    :param processor: Optional fitted :class:`~nextrec.data.preprocessor.DataProcessor` for preprocessing.
-
-    Examples
-    --------
-    >>> loader = RecDataLoader(
-    ...     dense_features=dense_features,
-    ...     sparse_features=sparse_features,
-    ...     sequence_features=sequence_features,
-    ...     target=['label'],
-    ...     processor=processor,
-    ... )
-    >>> dataloader = loader.create_dataloader(
-    ...     data=\"/path/to/data.csv\",
-    ...     batch_size=1024,
-    ...     load_full=False,
-    ...     chunk_size=20000,
-    ... )
-    """
-    
     def __init__(self,
                  dense_features: list[DenseFeature] | None = None,
                  sparse_features: list[SparseFeature] | None = None,
@@ -199,20 +134,8 @@ class RecDataLoader(FeatureSpecMixin):
                  target: list[str] | None | str = None,
                  id_columns: str | list[str] | None = None,
                  processor: Optional['DataProcessor'] = None):
-        """
-        Initialize the loader with feature/target definitions.
-
-        :param dense_features: Dense feature definitions (float).
-        :param sparse_features: Sparse feature definitions (int).
-        :param sequence_features: Sequence feature definitions (int, padded).
-        :param target: Single target name or list of names.
-        :param id_columns: Optional ID columns to append in output.
-        :param processor: Optional fitted ``DataProcessor`` for preprocessing.
-        """
-
         self.processor = processor
-        self._set_feature_config(dense_features, sparse_features, sequence_features)
-        self._set_target_config(target, id_columns)
+        self._set_feature_config(dense_features, sparse_features, sequence_features, target, id_columns)
 
     def create_dataloader(self,
                          data: Union[dict, pd.DataFrame, str, DataLoader],
@@ -220,79 +143,38 @@ class RecDataLoader(FeatureSpecMixin):
                          shuffle: bool = True,
                          load_full: bool = True,
                          chunk_size: int = 10000) -> DataLoader:
-        """
-        Build a ``DataLoader`` from in-memory data, file path, or an existing loader.
-
-        :param data: Dict/DataFrame (in-memory), path to CSV/Parquet file/dir, or an existing ``DataLoader``.
-        :param batch_size: Batch size for the returned ``DataLoader``.
-        :param shuffle: Shuffle flag passed to PyTorch ``DataLoader`` (for in-memory and streaming batches).
-        :param load_full: If ``True``, load all files into memory; if ``False``, stream with chunks.
-        :param chunk_size: Number of rows per chunk when ``load_full=False``.
-        :returns: A configured PyTorch ``DataLoader``.
-        """
         if isinstance(data, DataLoader):
             return data
-        
-        if isinstance(data, (str, os.PathLike)):
-            return self._create_from_path(data, batch_size, shuffle, load_full, chunk_size)
-        
-        if isinstance(data, (dict, pd.DataFrame)):
-            return self._create_from_memory(data, batch_size, shuffle)
-
-        raise ValueError(f"Unsupported data type: {type(data)}")
+        elif isinstance(data, (str, os.PathLike)):
+            return self._create_from_path(path=data, batch_size=batch_size, shuffle=shuffle, load_full=load_full, chunk_size=chunk_size)
+        elif isinstance(data, (dict, pd.DataFrame)):
+            return self._create_from_memory(data=data, batch_size=batch_size, shuffle=shuffle)
+        else:
+            raise ValueError(f"Unsupported data type: {type(data)}")
     
     def _create_from_memory(self, 
                            data: Union[dict, pd.DataFrame],
                            batch_size: int,
                            shuffle: bool) -> DataLoader:
-        """
-        Convert in-memory data (dict/DataFrame) into tensors and wrap with ``DataLoader``.
-
-        :param data: Dict or DataFrame containing feature/target columns.
-        :param batch_size: Batch size.
-        :param shuffle: Whether to shuffle batches.
-        :returns: A ``DataLoader`` backed by ``TensorDataset``.
-        """
-
         raw_data = data
 
         if self.processor is not None:
-            assert self.processor.is_fitted, "DataProcessor must be fitted before using in RecDataLoader"
+            if not self.processor.is_fitted:
+                raise ValueError("DataProcessor must be fitted before transforming data in memory")
             data = self.processor.transform(data, return_dict=True)
-
-        tensors = build_tensors_from_data(
-            data=data,
-            raw_data=raw_data,
-            features=self.all_features,
-            target_columns=self.target_columns,
-            id_columns=self.id_columns,
-            on_missing_feature="raise",
-        )
-
-        assert tensors is not None, "No tensors were created from provided data."
-
-        dataset = TensorDataset(*tensors)
-        return DataLoader(dataset, batch_size=batch_size, shuffle=shuffle)
+        tensors = build_tensors_from_data(data=data,raw_data=raw_data, features=self.all_features, target_columns=self.target_columns, id_columns=self.id_columns,)
+        if tensors is None:
+            raise ValueError("No valid tensors could be built from the provided data.")
+        dataset = TensorDictDataset(tensors)
+        return DataLoader(dataset, batch_size=batch_size, shuffle=shuffle, collate_fn=collate_fn)
     
     def _create_from_path(self,
                          path: str,
                          batch_size: int,
                          shuffle: bool,
                          load_full: bool,
-                         chunk_size: int) -> DataLoader:
-        """
-        Build a ``DataLoader`` from a CSV/Parquet file or directory.
-
-        :param path: File path or directory containing homogeneous CSV/Parquet files.
-        :param batch_size: Batch size.
-        :param shuffle: Shuffle flag.
-        :param load_full: If ``True``, load all rows into memory; otherwise stream.
-        :param chunk_size: Chunk rows when streaming.
-        :returns: A ``DataLoader`` (in-memory or streaming).
-        """
-
+                         chunk_size: int = 10000) -> DataLoader:
         file_paths, file_type = resolve_file_paths(str(Path(path)))
-        
         # Load full data into memory
         if load_full:
             dfs = []
@@ -306,20 +188,12 @@ class RecDataLoader(FeatureSpecMixin):
                     df = read_table(file_path, file_type)
                     dfs.append(df)
                 except MemoryError as exc:
-                    raise MemoryError(
-                        f"Out of memory while reading {file_path}. "
-                        f"Consider using load_full=False with streaming."
-                    ) from exc
-            
+                    raise MemoryError(f"Out of memory while reading {file_path}. Consider using load_full=False with streaming.") from exc
             try:
                 combined_df = pd.concat(dfs, ignore_index=True)
             except MemoryError as exc:
-                raise MemoryError(
-                    f"Out of memory while concatenating loaded data (approx {total_bytes / (1024**3):.2f} GB). "
-                    f"Use load_full=False to stream or reduce chunk_size."
-                ) from exc
-
-            return self._create_from_memory(combined_df, batch_size, shuffle)
+                raise MemoryError(f"Out of memory while concatenating loaded data (approx {total_bytes / (1024**3):.2f} GB). Use load_full=False to stream or reduce chunk_size.") from exc
+            return self._create_from_memory(combined_df, batch_size, shuffle,)
         else:
             return self._load_files_streaming(file_paths, file_type, batch_size, chunk_size, shuffle)
 
@@ -329,26 +203,10 @@ class RecDataLoader(FeatureSpecMixin):
                              batch_size: int,
                              chunk_size: int,
                              shuffle: bool) -> DataLoader:
-        """
-        Create a streaming ``DataLoader`` that yields chunked tensors from files.
-
-        :param file_paths: Ordered list of file paths.
-        :param file_type: ``\"csv\"`` or ``\"parquet\"``.
-        :param batch_size: Batch size for the outer ``DataLoader``.
-        :param chunk_size: Number of rows per chunk when reading files.
-        :returns: Streaming ``DataLoader`` with custom ``collate_fn``.
-        """
-        
         if shuffle:
-            logging.warning(colorize("Shuffle is ignored in streaming mode (IterableDataset).", "yellow"))
-
+            logging.warning("Shuffle is ignored in streaming mode (IterableDataset).")
         if batch_size != 1:
-            logging.warning(colorize(
-                "Streaming mode enforces batch_size=1; tune chunk_size to control memory/throughput.",
-                "yellow",
-            ))
-        effective_batch_size = 1
-
+            logging.warning("Streaming mode enforces batch_size=1; tune chunk_size to control memory/throughput.")
         dataset = FileDataset(
             file_paths=file_paths,
             dense_features=self.dense_features,
@@ -360,41 +218,24 @@ class RecDataLoader(FeatureSpecMixin):
             file_type=file_type,
             processor=self.processor
         )
-        
-        return DataLoader(dataset, batch_size=effective_batch_size, collate_fn=collate_fn)
+        return DataLoader(dataset, batch_size=1, collate_fn=collate_fn)
 
 def _normalize_sequence_column(column, feature: SequenceFeature) -> np.ndarray:
-    """
-    Normalize a raw sequence column into a padded int64 ``ndarray``.
-
-    :param column: Sequence column from DataFrame/dict; can be Series, list, or ndarray.
-    :param feature: Sequence feature definition providing ``max_len`` and optional ``padding_idx``.
-    :returns: 2-D numpy array (batch, seq_len) with dtype ``int64``.
-    """
     if isinstance(column, pd.Series):
         column = column.tolist()
-
     if isinstance(column, (list, tuple)):
         column = np.array(column, dtype=object)
-
     if not isinstance(column, np.ndarray):
         column = np.array([column], dtype=object)
-
     if column.ndim == 0:
         column = column.reshape(1)
-
     if column.dtype == object and any(isinstance(v, str) for v in column.ravel()):
-        raise TypeError(
-            f"Sequence feature '{feature.name}' expects numeric sequences; found string values."
-        )
-
+        raise TypeError(f"Sequence feature '{feature.name}' expects numeric sequences; found string values.")
     if column.dtype == object and len(column) > 0 and isinstance(column[0], (list, tuple, np.ndarray)):
         sequences = []
         for seq in column:
             if isinstance(seq, str):
-                raise TypeError(
-                    f"Sequence feature '{feature.name}' expects numeric sequences; found string values."
-                )
+                raise TypeError(f"Sequence feature '{feature.name}' expects numeric sequences; found string values.")
             if isinstance(seq, (list, tuple, np.ndarray)):
                 arr = np.asarray(seq, dtype=np.int64)
             else:
@@ -415,7 +256,6 @@ def _normalize_sequence_column(column, feature: SequenceFeature) -> np.ndarray:
         column = np.stack(padded)
     elif column.ndim == 1:
         column = column.reshape(-1, 1)
-
     return np.asarray(column, dtype=np.int64)
 
 
@@ -424,67 +264,38 @@ def build_tensors_from_data(  # noqa: C901
     raw_data: dict | pd.DataFrame,
     features: list,
     target_columns: list[str],
-    id_columns: list[str],
-    on_missing_feature: str = "raise",
-) -> tuple | None:
-    """
-    Shared routine to convert structured data into a tuple of tensors.
-
-    :param data: Preprocessed data (dict or DataFrame) used to fetch model inputs/labels.
-    :param raw_data: Original data, used for untouched ID columns.
-    :param features: Ordered list of feature definitions.
-    :param target_columns: Target/label column names.
-    :param id_columns: Extra ID column names to append at the end of the tensor tuple.
-    :param on_missing_feature: ``\"warn\"`` to skip missing feature with warning, ``\"raise\"`` to error.
-    :returns: Tuple of tensors following the order of ``features`` + targets (+ ids) or ``None`` if empty.
-    """
-    tensors: list[torch.Tensor] = []
-
+    id_columns: list[str]
+) -> dict | None:
+    feature_tensors: dict[str, torch.Tensor] = {}
     for feature in features:
         column = get_column_data(data, feature.name)
         if column is None:
-            if on_missing_feature == "warn":
-                logging.warning(colorize(f"Feature column '{feature.name}' not found in data", "yellow"))
-                continue
-            raise AssertionError(f"Feature column {feature.name} not found in data.")
-
+            raise ValueError(f"Feature column '{feature.name}' not found in data")
         if isinstance(feature, SequenceFeature):
             tensor = torch.from_numpy(_normalize_sequence_column(column, feature))
         elif isinstance(feature, DenseFeature):
             tensor = torch.from_numpy(np.asarray(column, dtype=np.float32))
         else:
             tensor = torch.from_numpy(np.asarray(column, dtype=np.int64))
-
-        tensors.append(tensor)
-
-    label_tensors = []
+        feature_tensors[feature.name] = tensor
+    label_tensors = None
     if target_columns:
+        label_tensors = {}
         for target_name in target_columns:
             column = get_column_data(data, target_name)
-            assert column is not None, f"Target column '{target_name}' not found in data."
-
+            if column is None:
+                continue
             label_tensor = torch.from_numpy(np.asarray(column, dtype=np.float32))
-
-            if label_tensor.dim() == 1:
-                label_tensor = label_tensor.view(-1, 1)
-            elif label_tensor.dim() == 2 and label_tensor.shape[0] == 1 and label_tensor.shape[1] > 1:
+            if label_tensor.dim() == 2 and label_tensor.shape[0] == 1 and label_tensor.shape[1] > 1:
                 label_tensor = label_tensor.t()
-
-            label_tensors.append(label_tensor)
-
-    if label_tensors:
-        if len(label_tensors) == 1 and label_tensors[0].shape[1] > 1:
-            y_tensor = label_tensors[0]
-        else:
-            y_tensor = torch.cat(label_tensors, dim=1)
-
-        if y_tensor.shape[1] == 1:
-            y_tensor = y_tensor.squeeze(1)
-
-        tensors.append(y_tensor)
-
+            if label_tensor.shape[1:] == (1,):
+                label_tensor = label_tensor.squeeze(1)
+            label_tensors[target_name] = label_tensor
+        if not label_tensors:
+            label_tensors = None
+    id_tensors = None
     if id_columns:
-        id_arrays = []
+        id_tensors = {}
         for id_col in id_columns:
             column = get_column_data(raw_data, id_col)
             if column is None:
@@ -494,20 +305,8 @@ def build_tensors_from_data(  # noqa: C901
             try:
                 id_arr = np.asarray(column, dtype=np.int64)
             except Exception as exc:
-                raise TypeError(
-                    f"ID column '{id_col}' must contain numeric values. "
-                    f"Received dtype={np.asarray(column).dtype}, error: {exc}"
-                ) from exc
-            id_arrays.append(id_arr)
-
-        combined_ids = np.column_stack(id_arrays)
-        tensors.append(torch.from_numpy(combined_ids))
-
-    if not tensors:
+                raise TypeError( f"ID column '{id_col}' must contain numeric values. Received dtype={np.asarray(column).dtype}, error: {exc}") from exc
+            id_tensors[id_col] = torch.from_numpy(id_arr)
+    if not feature_tensors:
         return None
-
-    return tuple(tensors)
-
-
-# Backward compatible alias
-_build_tensors_from_data = build_tensors_from_data
+    return {"features": feature_tensors, "labels": label_tensors, "ids": id_tensors}