proxyml 0.1.0__tar.gz → 0.1.2__tar.gz

{proxyml-0.1.0 → proxyml-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proxyml
-Version: 0.1.0
+Version: 0.1.2
 Summary: Python SDK for calling the ProxyML API
 Author-email: ProxyML <contact@proxyml.ai>
 License:                                  Apache License

{proxyml-0.1.0 → proxyml-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "proxyml"
-version = "0.1.0"
+version = "0.1.2"
 description = "Python SDK for calling the ProxyML API"
 readme = "README.md"
 license = {file = "LICENSE"}

{proxyml-0.1.0 → proxyml-0.1.2}/src/proxyml/__init__.py

@@ -5,6 +5,7 @@ from proxyml.client import (
     predict,
     find_counterfactual,
     interpret_counterfactual,
+    get_feature_importances,
 )
 from proxyml.schema import (
     get_schema,

{proxyml-0.1.0 → proxyml-0.1.2}/src/proxyml/client.py

@@ -44,6 +44,15 @@ def put(endpoint: str, payload: dict) -> requests.models.Response:
     return r
 
 
+def get(endpoint: str, params: dict) -> requests.models.Response:
+    r = requests.get(
+        url=f'{PROXYML_BASE_URL}{endpoint}',
+        headers=_headers(),
+        params=params
+    )
+    return r
+
+
 def put_schema(schema: dict):
     r = put(endpoint='/schema', payload=schema)
     if r.status_code == 200:
@@ -127,10 +136,10 @@ def predict(samples: list, version: int | None):  # Defaults to latest version i
 def find_counterfactual(sample, target, n_neighbors: int = 10000, perturbation_scale: float = 0.1, version: int | None = None, as_df: bool = True):
     payload = {
         'instance': sample,
-        'target_label': target,
+        'target_label': target.item() if hasattr(target, 'item') else target,  # handle numpy scalars
         'n_neighbors': n_neighbors,
         'perturbation_scale': perturbation_scale,
-    }
+    }    
     if version:  # Also rejects version 0 (versions start at 1)
         payload['version'] = version 
     r = post(endpoint='/explain/counterfactual', payload=payload)
@@ -188,3 +197,16 @@ def interpret_counterfactual(
             f"This may indicate the surrogate model does not fully capture "
             f"the original model's decision boundary in this region."
         )
+
+
+def get_feature_importances(version: int | None = None, ):
+    params = {'version': version} if version else {}
+    r = get(endpoint='/explain/importance', params=params)
+    if r.status_code == 200:
+        return r.json()
+    logger.error(
+        "Feature importances failed with status %s: %s",
+        r.status_code,
+        r.text,
+    )
+    return None 

{proxyml-0.1.0 → proxyml-0.1.2}/src/proxyml.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proxyml
-Version: 0.1.0
+Version: 0.1.2
 Summary: Python SDK for calling the ProxyML API
 Author-email: ProxyML <contact@proxyml.ai>
 License:                                  Apache License