Fix filenames accordingly.

mosaicml · Dec 15, 2023 · 7cf0ef3 · 7cf0ef3
1 parent e2f3a83
commit 7cf0ef3
Show file tree

Hide file tree

Showing 8 changed files with 7 additions and 7 deletions.
diff --git a/streaming/format/__init__.py b/streaming/format/__init__.py
@@ -8,7 +8,7 @@
 from streaming.format.index import get_index_basename
 from streaming.format.json import JSONShard, JSONWriter
 from streaming.format.mds import MDSShard, MDSWriter
-from streaming.format.reader import FileInfo, Shard
+from streaming.format.shard import FileInfo, Shard
 from streaming.format.xsv import CSVShard, CSVWriter, TSVShard, TSVWriter, XSVShard, XSVWriter
 
 __all__ = [

diff --git a/streaming/format/json/__init__.py b/streaming/format/json/__init__.py
@@ -3,7 +3,7 @@
 
 """Streaming JSON shards."""
 
-from streaming.format.json.reader import JSONShard
+from streaming.format.json.shard import JSONShard
 from streaming.format.json.writer import JSONWriter
 
 __all__ = ['JSONShard', 'JSONWriter']
diff --git a/streaming/format/json/reader.py → streaming/format/json/shard.py b/streaming/format/json/reader.py → streaming/format/json/shard.py
@@ -11,7 +11,7 @@
 import numpy as np
 from typing_extensions import Self
 
-from streaming.format.reader import DualShard, FileInfo
+from streaming.format.shard import DualShard, FileInfo
 
 __all__ = ['JSONShard']
 

diff --git a/streaming/format/mds/__init__.py b/streaming/format/mds/__init__.py
@@ -3,7 +3,7 @@
 
 """MDS shards."""
 
-from streaming.format.mds.reader import MDSShard
+from streaming.format.mds.shard import MDSShard
 from streaming.format.mds.writer import MDSWriter
 
 __all__ = ['MDSShard', 'MDSWriter']
diff --git a/streaming/format/mds/reader.py → streaming/format/mds/shard.py b/streaming/format/mds/reader.py → streaming/format/mds/shard.py
@@ -11,7 +11,7 @@
 from typing_extensions import Self
 
 from streaming.format.mds.encodings import is_mds_encoding_safe, mds_decode
-from streaming.format.reader import FileInfo, MonoShard
+from streaming.format.shard import FileInfo, MonoShard
 
 __all__ = ['MDSShard']
 

diff --git a/streaming/format/reader.py → streaming/format/shard.py b/streaming/format/reader.py → streaming/format/shard.py
diff --git a/streaming/format/xsv/__init__.py b/streaming/format/xsv/__init__.py
@@ -3,7 +3,7 @@
 
 """Streaming XSV shards, with specializations for CSV and TSV."""
 
-from streaming.format.xsv.reader import CSVShard, TSVShard, XSVShard
+from streaming.format.xsv.shard import CSVShard, TSVShard, XSVShard
 from streaming.format.xsv.writer import CSVWriter, TSVWriter, XSVWriter
 
 __all__ = ['CSVShard', 'CSVWriter', 'TSVShard', 'TSVWriter', 'XSVShard', 'XSVWriter']
diff --git a/streaming/format/xsv/reader.py → streaming/format/xsv/shard.py b/streaming/format/xsv/reader.py → streaming/format/xsv/shard.py
@@ -10,7 +10,7 @@
 import numpy as np
 from typing_extensions import Self
 
-from streaming.format.reader import DualShard, FileInfo
+from streaming.format.shard import DualShard, FileInfo
 from streaming.format.xsv.encodings import xsv_decode
 
 __all__ = ['XSVShard', 'CSVShard', 'TSVShard']