datfid 0.1.14__py3-none-any.whl → 0.1.16__py3-none-any.whl

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.dist-info → datfid-0.1.16.dist-info}/METADATA

@@ -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.16.dist-info/RECORD

@@ -0,0 +1,7 @@
+datfid/__init__.py,sha256=PojdUQsimZvawR8lgqVpzs3EkRtkA7iwuDk7fovJF54,34
+datfid/client.py,sha256=uza0OZBtm5aQF0xdFF2TIBUOYeQ9zYmX-hS8-OBXWqw,8267
+datfid-0.1.16.dist-info/licenses/LICENSE,sha256=5b2JLb1-P8Nt919phjUmP0umS82SxGMlIF2QFn9ILgM,1084
+datfid-0.1.16.dist-info/METADATA,sha256=frz6VkxxFwrNeY7-ZbN2U4H7nCwN3-DIH9WkEQyHJtU,5146
+datfid-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+datfid-0.1.16.dist-info/top_level.txt,sha256=VFG23u9NMNQJ8S_IFfW-S5d8oSUd71SFA346ovh_uxA,7
+datfid-0.1.16.dist-info/RECORD,,

datfid-0.1.14.dist-info/RECORD

@@ -1,7 +0,0 @@
-datfid/__init__.py,sha256=PojdUQsimZvawR8lgqVpzs3EkRtkA7iwuDk7fovJF54,34
-datfid/client.py,sha256=U2fuj2nPadXAQ0lGdVinu0ShWdeC4pHWwK1nNhPQD0E,8045
-datfid-0.1.14.dist-info/licenses/LICENSE,sha256=5b2JLb1-P8Nt919phjUmP0umS82SxGMlIF2QFn9ILgM,1084
-datfid-0.1.14.dist-info/METADATA,sha256=KHLb4wwu5ZIaFfI5R_GMa-_nhVFXo053t3CqYi_C6VE,5146
-datfid-0.1.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-datfid-0.1.14.dist-info/top_level.txt,sha256=VFG23u9NMNQJ8S_IFfW-S5d8oSUd71SFA346ovh_uxA,7
-datfid-0.1.14.dist-info/RECORD,,