proxyml 0.1.0__tar.gz → 0.1.1__tar.gz

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

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

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

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

{proxyml-0.1.0 → proxyml-0.1.1}/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.1}/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:
@@ -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.1}/src/proxyml.egg-info/PKG-INFO

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