remove moved class from import

This commit is contained in:
Wing Lian
2024-12-30 13:17:11 -05:00
parent 06370b386a
commit 385c60cd9b

View File

@@ -7,5 +7,4 @@ from .batching import ( # noqa: F401
PretrainingBatchSamplerDataCollatorForSeq2Seq,
V2BatchSamplerDataCollatorForSeq2Seq,
)
from .kd import DataCollatorForKD # noqa: F401
from .mamba import MambaDataCollator # noqa: F401