datfid 0.1.14__tar.gz → 0.1.16__tar.gz

{datfid-0.1.14 → datfid-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datfid
-Version: 0.1.14
+Version: 0.1.16
 Summary: SDK to access the DATFID API hosted on Hugging Face Spaces
 Author: DATFID
 Author-email: igor.schapiro@datfid.com

{datfid-0.1.14 → datfid-0.1.16}/datfid/client.py

@@ -145,7 +145,7 @@ class DATFIDClient:
             self.logger.error(f"Forecast generation failed: {str(e)}")
             raise
 
-def fit_model_ind(
+    def fit_model_ind(
         self,
         df: pd.DataFrame,
         id_col: str,
@@ -156,7 +156,7 @@ def fit_model_ind(
         current_features: Optional[list] = None,
         filter_by_significance: bool = False,
         meanvar_test: bool = False
-    ) -> SimpleNamespace:
+    ) -> Dict[str, SimpleNamespace]:
         """
         Fit a model individual by individual using the DATFID API.
         
@@ -200,10 +200,12 @@ def fit_model_ind(
         if response.status_code != 200:
             raise Exception(f"Model fit failed: {response.text}")
             
-        result_dict = response.json()
-        return SimpleNamespace(**result_dict)
+        raw = response.json()  # { "<id>": { ... per-id result ... }, ... }
+        # Wrap each per-id result into a SimpleNamespace for dot access:
+        result_per_id = {str(k): SimpleNamespace(**v) for k, v in raw.items()}
+        return result_per_id  # Dict[str, SimpleNamespace]
 
-def forecast_model_ind(
+    def forecast_model_ind(
         self,
         df_forecast: pd.DataFrame
     ) -> pd.DataFrame:

{datfid-0.1.14 → datfid-0.1.16}/datfid.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datfid
-Version: 0.1.14
+Version: 0.1.16
 Summary: SDK to access the DATFID API hosted on Hugging Face Spaces
 Author: DATFID
 Author-email: igor.schapiro@datfid.com

{datfid-0.1.14 → datfid-0.1.16}/setup.py

@@ -6,7 +6,7 @@ from setuptools import setup, find_packages
 # 3) delete folder datfid.egg-info
 # 4) delete older files from dist folder
 # 5) in terminal: python setup.py sdist bdist_wheel
-# 6) in terminal: twine upload --repository testpypi dist/*
+# 6) in terminal: twine upload --repository pypi dist/*
 # 7) in hugging face delete older files from dist folder
 # 8) in hugging face upload updated files
 # 9) in terminal uninstall older version of datfid: pip uninstall datfid
@@ -17,7 +17,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="datfid",
-    version="0.1.14",
+    version="0.1.16",
     description="SDK to access the DATFID API hosted on Hugging Face Spaces",
     long_description=long_description,
     long_description_content_type="text/markdown",  # Important!