remove moved class from import

2024-12-30 13:17:11 -05:00
parent 327739c9e3
commit 723b0a2dee
1 changed files with 0 additions and 1 deletions
--- a/src/axolotl/utils/collators/init.py
+++ b/src/axolotl/utils/collators/init.py
@@ -7,5 +7,4 @@ from .batching import (  # noqa: F401
    PretrainingBatchSamplerDataCollatorForSeq2Seq,
    V2BatchSamplerDataCollatorForSeq2Seq,
 )
-from .kd import DataCollatorForKD  # noqa: F401
 from .mamba import MambaDataCollator  # noqa: F401