datamule 0.421__tar.gz → 0.423__tar.gz

{datamule-0.421 → datamule-0.423}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: datamule
-Version: 0.421
+Version: 0.423
 Summary: Making it easier to use SEC filings.
 Home-page: https://github.com/john-friedman/datamule-python
 Author: John Friedman

{datamule-0.421 → datamule-0.423}/datamule/portfolio.py

@@ -4,13 +4,23 @@ from concurrent.futures import ProcessPoolExecutor
 from .submission import Submission
 
 class Portfolio:
-    def __init__(self, path):
+    @classmethod
+    def create(cls, path):
+        # This method handles the process pool lifecycle
+        with ProcessPoolExecutor() as executor:
+            portfolio = cls(path, executor)
+            return portfolio
+
+    def __init__(self, path, executor=None):
         self.path = Path(path)
         folders = [f for f in self.path.iterdir() if f.is_dir()]
         print(f"Loading {len(folders)} submissions")
-        # Load submissions in parallel
-        with ProcessPoolExecutor() as executor:
-            # Show progress while loading
+        
+        if executor is None:
+            # Fall back to sequential loading if no executor
+            self.submissions = [Submission(f) for f in tqdm(folders, desc="Loading submissions")]
+        else:
+            # Use provided executor for parallel loading
             self.submissions = list(tqdm(
                 executor.map(Submission, folders),
                 total=len(folders),

{datamule-0.421 → datamule-0.423}/datamule.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: datamule
-Version: 0.421
+Version: 0.423
 Summary: Making it easier to use SEC filings.
 Home-page: https://github.com/john-friedman/datamule-python
 Author: John Friedman

{datamule-0.421 → datamule-0.423}/setup.py

@@ -3,15 +3,7 @@ from pathlib import Path
 import platform
 import os
 from setuptools import find_namespace_packages  # Add this import
-
-
-# First, try to install Cython if it's not available
-try:
-    from Cython.Build import cythonize
-except ImportError:
-    from setuptools import dist
-    dist.Distribution().fetch_build_eggs(['Cython'])
-    from Cython.Build import cythonize
+from Cython.Build import cythonize
 
 
 # Platform-specific settings
@@ -63,7 +55,7 @@ extras["all"] = list(all_dependencies)
 setup(
     name="datamule",
     author="John Friedman",
-    version="0.421",
+    version="0.423",
     description="Making it easier to use SEC filings.",
     packages=find_namespace_packages(include=['datamule*']),
     url="https://github.com/john-friedman/datamule-python",

{datamule-0.421 → datamule-0.423}/datamule.egg-info/requires.txt

@@ -11,11 +11,11 @@ pytz
 zstandard
 
 [all]
-pandas
-openai
+psutil
 google-generativeai
+openai
 flask
-psutil
+pandas
 
 [dataset_builder]
 pandas