Source code for rllm.transforms.table_transforms.col_transform

from abc import ABC, abstractmethod
import copy




class ColTransform(ABC):
    r"""An abstract base class for transforming individual column features
    in table data. It provides a common interface for all column
    transformation operations. It ensures that the data is shallow-copied
    to prevent in-place modifications.
    """

    def __call__(self, data):
        # Shallow-copy the data so that we prevent in-place data modification.
        return self.forward(copy.copy(data))

    @abstractmethod
    def forward(self, data):
        raise NotImplementedError

    def __repr__(self) -> str:
        return f"{self.__class__.__name__}()"