datfid 0.1.18__tar.gz → 0.1.19__tar.gz

{datfid-0.1.18 → datfid-0.1.19}/PKG-INFO

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

{datfid-0.1.18 → datfid-0.1.19}/datfid/client.py

@@ -16,8 +16,24 @@ class FitResult(SimpleNamespace):
     
     @property
     def id(self):
-        """Return list of all individual IDs"""
-        return list(self.keys())
+        import pandas as pd
+        if hasattr(self, "df") and isinstance(self.df, pd.DataFrame) and "ID" in self.df.columns:
+            return pd.DataFrame(self.df["ID"].astype(str).unique())
+        return pd.DataFrame([])
+    
+    @property
+    def ID(self):
+        import pandas as pd
+        if hasattr(self, "df") and isinstance(self.df, pd.DataFrame) and "ID" in self.df.columns:
+            return pd.DataFrame(self.df["ID"].astype(str).unique())
+        return pd.DataFrame([])
+    
+    @property
+    def Id(self):
+        import pandas as pd
+        if hasattr(self, "df") and isinstance(self.df, pd.DataFrame) and "ID" in self.df.columns:
+            return pd.DataFrame(self.df["ID"].astype(str).unique())
+        return pd.DataFrame([])
 
     @staticmethod
     def _df4(rows_list):
@@ -43,6 +59,20 @@ class FitResult(SimpleNamespace):
             kwargs["df"] = pd.DataFrame(kwargs["df"])
         super().__init__(**kwargs)
 
+class FitResultDict(dict):
+    @property
+    def id(self):
+        return pd.DataFrame({"ID": list(self.keys())})
+    
+    @property
+    def ID(self):
+        return pd.DataFrame({"ID": list(self.keys())})
+    
+    @property
+    def Id(self):
+        return pd.DataFrame({"ID": list(self.keys())})
+
+
 class DATFIDClient:
     def __init__(self, token: str):
         self.api_url = "https://datfid-org-datfid-sdk.hf.space/"
@@ -190,7 +220,7 @@ class DATFIDClient:
         current_features: Optional[list] = None,
         filter_by_significance: bool = False,
         meanvar_test: bool = False
-    ) -> Dict[str, FitResult]:
+    ) -> FitResultDict:
         """
         Fit a model individual by individual using the DATFID API.
         
@@ -234,10 +264,10 @@ class DATFIDClient:
         if response.status_code != 200:
             raise Exception(f"Model fit failed: {response.text}")
             
-        raw = response.json()  # { "<id>": { ... per-id result ... }, ... }
+        raw = response.json() 
         # Wrap each per-id result into a SimpleNamespace for dot access:
-        result_per_id = {str(k): FitResult(**v) for k, v in raw.items()}
-        return result_per_id  # Dict[str, SimpleNamespace]
+        result_per_id = FitResultDict({str(k): FitResult(**v) for k, v in raw.items()})
+        return result_per_id  # FitResultDict[str, SimpleNamespace]
 
     def forecast_model_ind(
         self,

{datfid-0.1.18 → datfid-0.1.19}/datfid.egg-info/PKG-INFO

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

{datfid-0.1.18 → datfid-0.1.19}/setup.py

@@ -17,7 +17,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="datfid",
-    version="0.1.18",
+    version="0.1.19",
     description="SDK to access the DATFID API hosted on Hugging Face Spaces",
     long_description=long_description,
     long_description_content_type="text/markdown",  # Important!