update

mosaicml · Oct 29, 2024 · aa3c798 · aa3c798
1 parent f588bd5
commit aa3c798
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/streaming/base/util.py b/streaming/base/util.py
@@ -20,7 +20,7 @@
 
 import torch.distributed as dist
 
-from streaming.base.constant import SHM_TO_CLEAN, BARRIER_FILELOCK, CACHE_FILELOCK
+from streaming.base.constant import BARRIER_FILELOCK, CACHE_FILELOCK, SHM_TO_CLEAN
 from streaming.base.distributed import get_local_rank, maybe_init_dist
 from streaming.base.format.index import get_index_basename
 from streaming.base.shared.prefix import _get_path