Source code for mlx_graphs.transforms.base_transform

import copy
from abc import ABC, abstractmethod
from typing import Any




class BaseTransform(ABC):
    r"""An abstract class for creating transformations
    Transforms are a way to modify and customize
    GraphData by implicitly passing them as arguments
    in mlx_graphs.dataset.Dataset


    """

    def __call__(self, data: Any) -> Any:
        return self.process(copy.copy(data))

    @abstractmethod
    def process(self, data: Any) -> Any:
        pass

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