fix lint

allenai · Dec 3, 2024 · c1f7d44 · c1f7d44
1 parent bed869f
commit c1f7d44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/examples/llama/train.py b/src/examples/llama/train.py
@@ -17,7 +17,7 @@
     NumpyDatasetType,
     TokenizerConfig,
 )
-from olmo_core.distributed.parallel import DataParallelType, get_dp_process_group
+from olmo_core.distributed.parallel import DataParallelType
 from olmo_core.nn.transformer import TransformerConfig, TransformerDataParallelConfig
 from olmo_core.optim import AdamWConfig, CosWithWarmup, OptimGroupOverride
 from olmo_core.train import (