Added DataLoaderConfig to Namespace

kamilkrukowski · Mar 7, 2023 · c79d60d · c79d60d
1 parent bb0836b
commit c79d60d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "edgar_doc_parser"
-version = "0.2.2"
+version = "0.2.2-1"
 authors = [
   { name="Kamil Krukowski", email="[email protected]" },
   { name="Yiting Li"},

diff --git a/src/edgar/__init__.py b/src/edgar/__init__.py
@@ -5,7 +5,7 @@
 from .document import DocumentType
 from .wrappers import load_files, get_files, read_file
 from . import pipeline
-from .dataloader import DataLoader
+from .dataloader import DataLoader, DataLoaderConfig
 
 module_name = 'edgar'
 
@@ -14,6 +14,8 @@
 _Metadata.__module__ = module_name
 DocumentType.__module__ = module_name
 DataLoader.__module__ = module_name
+DataLoaderConfig.__module__ = module_name
+
 
 _, _, _ = load_files, get_files, read_file