include __version__ in all

mosaicml · Jul 31, 2024 · 0b84d4c · 0b84d4c
1 parent ac1dd82
commit 0b84d4c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llmfoundry/__init__.py b/llmfoundry/__init__.py
@@ -36,7 +36,6 @@
 
 logger.addFilter(new_files_warning_filter)
 
-from llmfoundry._version import __version__
 from llmfoundry import (
     algorithms,
     callbacks,
@@ -51,6 +50,7 @@
     tokenizers,
     utils,
 )
+from llmfoundry._version import __version__
 from llmfoundry.data import StreamingFinetuningDataset, StreamingTextDataset
 from llmfoundry.eval import InContextLearningDataset, InContextLearningMetric
 from llmfoundry.models.hf import ComposerHFCausalLM
@@ -64,6 +64,7 @@
 from llmfoundry.optim import DecoupledLionW
 
 __all__ = [
+    '__version__',
     'StreamingFinetuningDataset',
     'StreamingTextDataset',
     'InContextLearningDataset',