Apply isort and black reformatting

Signed-off-by: huvunvidia <huvunvidia@users.noreply.github.com>
NVIDIA · Nov 15, 2024 · a8ba110 · a8ba110
1 parent 48e4e6c
commit a8ba110
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/nemo/collections/llm/__init__.py b/nemo/collections/llm/__init__.py
@@ -27,12 +27,6 @@
     PreTrainingDataModule,
     SquadDataModule,
 )
-from nemo.collections.llm.t5.data import (
-    FineTuningDataModule as T5FineTuningDataModule,
-    MockDataModule as T5MockDataModule,
-    PreTrainingDataModule as T5PreTrainingDataModule,
-    SquadDataModule as T5SquadDataModule,
-)
 from nemo.collections.llm.gpt.data.api import dolly, hf_dataset, mock, squad
 from nemo.collections.llm.gpt.model import (
     Baichuan2Config,
@@ -118,6 +112,10 @@
     gpt_forward_step,
 )
 from nemo.collections.llm.quantization import Quantizer, get_calib_data_iter
+from nemo.collections.llm.t5.data import FineTuningDataModule as T5FineTuningDataModule
+from nemo.collections.llm.t5.data import MockDataModule as T5MockDataModule
+from nemo.collections.llm.t5.data import PreTrainingDataModule as T5PreTrainingDataModule
+from nemo.collections.llm.t5.data import SquadDataModule as T5SquadDataModule
 from nemo.collections.llm.t5.model import T5Config, T5Model, t5_data_step, t5_forward_step
 
 __all__ = [

diff --git a/nemo/collections/llm/t5/data/__init__.py b/nemo/collections/llm/t5/data/__init__.py
@@ -1,6 +1,6 @@
 from nemo.collections.llm.t5.data.fine_tuning import FineTuningDataModule
+from nemo.collections.llm.t5.data.mock import MockDataModule
 from nemo.collections.llm.t5.data.pre_training import PreTrainingDataModule
 from nemo.collections.llm.t5.data.squad import SquadDataModule
-from nemo.collections.llm.t5.data.mock import MockDataModule
 
 __all__ = ["FineTuningDataModule", "PreTrainingDataModule", "SquadDataModule", "MockDataModule"]