fixed main import

allenai · Aug 27, 2024 · f34cd37 · f34cd37
1 parent 3934449
commit f34cd37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/test_tokenizer.py b/tests/python/test_tokenizer.py
@@ -6,7 +6,7 @@
 
 import numpy
 import smart_open
-from dolma.cli.__main__ import main
+from dolma.cli.main import main
 from dolma.tokenizer import Tokenizer, tokenize_in_parallel
 from tokenizers import Tokenizer as BaseTokenizer
 from typing_extensions import TypedDict