diffab/utils/transforms/__init__.py

# Transforms
from .mask import MaskSingleCDR, MaskMultipleCDRs, MaskAntibody
from .merge import MergeChains
from .patch import PatchAroundAnchor

# Factory
from ._base import get_transform, Compose