From 723b0a2deeb63b4dc72133c7184092b0e17f973e Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 30 Dec 2024 13:17:11 -0500 Subject: [PATCH] remove moved class from import --- src/axolotl/utils/collators/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/axolotl/utils/collators/__init__.py b/src/axolotl/utils/collators/__init__.py index 649356e12..93502b67d 100644 --- 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