xplainable-client 1.2.1.post4__tar.gz → 1.2.2__tar.gz

{xplainable-client-1.2.1.post4 → xplainable_client-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xplainable-client
-Version: 1.2.1.post4
+Version: 1.2.2
 Summary: The client for persisting and deploying models to Xplainable cloud.
 Author-email: xplainable pty ltd <contact@xplainable.io>
 Classifier: Programming Language :: Python :: 3.8
@@ -11,14 +11,14 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: ipywidgets==8.0.6
-Requires-Dist: numpy==1.21.1
+Requires-Dist: numpy>=1.20.3
 Requires-Dist: pandas<=1.9.0,>=1.5.2
 Requires-Dist: pyperclip==1.8.2
 Requires-Dist: Requests==2.31.0
 Requires-Dist: scikit_learn
 Requires-Dist: setuptools
 Requires-Dist: urllib3==2.2.0
-Requires-Dist: xplainable==1.2.1
+Requires-Dist: xplainable
 
 
 <div align="center">

{xplainable-client-1.2.1.post4 → xplainable_client-1.2.2}/pyproject.toml

@@ -7,7 +7,7 @@ find = {}
 
 [project]
 name = "xplainable-client"
-version = "1.2.1-4"
+version = "1.2.2"
 authors = [
   { name="xplainable pty ltd", email="contact@xplainable.io" },
 ]
@@ -26,12 +26,12 @@ classifiers = [
 
 dependencies = [
 "ipywidgets==8.0.6",
-"numpy==1.21.1",
+"numpy>=1.20.3",
 "pandas>=1.5.2,<=1.9.0",
 "pyperclip==1.8.2",
 "Requests==2.31.0",
 "scikit_learn",
 "setuptools",
 "urllib3==2.2.0",
-"xplainable==1.2.1"
+"xplainable"
 ]

xplainable_client-1.2.2/xplainable_client/client/__init__.py

@@ -0,0 +1,33 @@
+from .datasets import *
+from functools import cached_property
+
+from ._models import *
+from ._misc import *
+from requests.packages.urllib3.util.retry import Retry
+from ._session import Session
+from ._misc import Misc
+from ._models import Models
+from ._preprocessing import Preprocessing
+from ._deployments import Deployments
+from ._gpt import GPT
+from ._collections import Collections
+from ._datasets import Datasets
+from ._inference import Inference
+
+
+class Client(Misc, Models, Preprocessing, GPT, Collections):
+
+    def __init__(self, api_key=None, hostname='https://api.xplainable.io', silent_mode=False):
+        self.session = Session(api_key=api_key, hostname=hostname, silent_mode=silent_mode)
+        Misc.__init__(self, self.session)
+        Models.__init__(self, self.session)
+        Preprocessing.__init__(self, self.session)
+        Deployments.__init__(self, self.session)
+        GPT.__init__(self, self.session)
+        Collections.__init__(self, self.session)
+        Datasets.__init__(self, self.session)
+        Inference.__init__(self, self.session)
+        
+
+
+

xplainable_client-1.2.2/xplainable_client/client/_client_cog_base.py

@@ -0,0 +1,7 @@
+from ._session import Session
+
+
+class Client_Cog:
+
+    def __init__(self, session: Session):
+        self.session = session

xplainable_client-1.2.2/xplainable_client/client/_collections.py

@@ -0,0 +1,213 @@
+from ._client_cog_base import *
+from ._models import Models
+
+
+class Collections(Client_Cog):
+
+    def create_collection(self, model_id: str, name: str, description: str):
+        """ Creates a new collection.
+        Args:
+            model_id (str): The model id.
+            name (str): The name of the collection.
+            description (str): The description of the collection.
+        Returns:
+            int: The id of the collection created.
+        """
+        payload = {
+            "name": name,
+            "description": description
+        }
+        
+        response = self.session._session.put(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/create-collection",
+            json=payload
+            )
+        data = self.session.get_response_content(response)
+        return data
+    
+    
+    def get_model_collections(self, model_id: str):
+        """ Gets all collections of a model.
+        Args:
+            model_id (str): The model id.
+        Returns:
+            dict: Dictionary of collections.
+        """
+        response = self.session._session.get(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections"
+            )
+        data = self.session.get_response_content(response)
+        return data
+    
+    def get_team_collections(self):
+        """ Gets all collections of a team.
+        Returns:
+            dict: Dictionary of collections.
+        """
+        response = self.session._session.get(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/collections"
+            )
+        data = self.session.get_response_content(response)
+        return data
+    
+
+    def edit_collection_name(self, model_id: str, collections_id: str, name: str):
+        """ Edits the name of a collection.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+            name (str): The new name of the collection.
+        Returns:
+            json: A json payload.
+        """
+        payload = {
+            "name": name
+        }
+
+        response = self.session._session.patch(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collections_id}/name",
+            json=payload
+            )
+        return response
+    
+
+    def edit_collection_description(self, model_id: str, collections_id: str, description: str):
+        """ Edits the description of a collection.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+            description (str): The new description of the collection.
+        Returns:
+            json: A json payload.
+        """
+        payload = {
+            "description": description
+        }
+
+        response = self.session._session.patch(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collections_id}/description",
+            json=payload
+            )
+        return response
+    
+
+    def delete_collection(self, model_id: str, collections_id: str):
+        """ Deletes a collection.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+        Returns:
+            json: A json payload.
+        """
+        response = self.session._session.delete(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collections_id}"
+            )
+        return response
+    
+
+    def delete_model_collections(self, model_id: str):
+        """ Deletes all collections of a model.
+        Args:
+            model_id (str): The model id.
+        Returns:
+            json: A json payload.
+        """
+        response = self.session._session.delete(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections"
+            )
+        return response
+    
+    
+    def restore_collection(self, model_id: str, collection_id: str):
+        """ Restores a collection.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+        Returns:
+            json: A json payload.
+        """
+        response = self.session._session.patch(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collection_id}/restore"
+            )
+        return response
+    
+
+    def get_collection(self, model_id: str, collection_id: str):
+        """ Gets a collection.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+        Returns:
+            dict: Dictionary of collection.
+        """
+        response = self.session._session.get(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collection_id}"
+            )
+        data = self.session.get_response_content(response)
+        return data
+    
+
+    #TODO: Implement.
+    # def add_scenario(self, model_id: str, version_id: str, collection_id: str):
+    #     try:
+    #         response = self.session._session.get(
+    #             url=f'{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/versions/{version_id}'
+    #         )
+    #         return self.session.get_response_content(response)
+    #     except Exception as e:
+    #         raise ValueError(f'Model with ID {model_id}:{version_id} does not exist')
+    #     return
+
+    # add_scenario
+
+
+    def edit_scenario_notes(self, model_id: str, collection_id: str, scenario_id: str, notes: str):
+        """ Edits the notes of a scenario.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+            scenario_id (str): The scenario id.
+            notes (str): The new notes of the scenario.
+        Returns:
+            json: A json payload.
+        """
+        payload = {
+            "notes": notes
+        }
+
+        response = self.session._session.patch(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collection_id}/scenarios/{scenario_id}/notes",
+            json=payload
+            )
+        return response
+    
+
+    def delete_scenario(self, model_id: str, collection_id: str, scenario_id: str):
+        """" Deletes a scenario.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+            scenario_id (str): The scenario id.
+        Returns:
+            json: A json payload.
+        """
+
+        response = self.session._session.delete(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collection_id}/scenarios/{scenario_id}"
+            )
+        return response
+    
+
+    def restore_scenario(self, model_id: str, collection_id: str, scenario_id: str):
+        """ Restores a scenario.
+        Args:
+            model_id (str): The model id.
+            collections_id (str): The collection id.
+            scenario_id (str): The scenario id.
+        Returns:
+            json: A json payload.
+        """
+        response = self.session._session.patch(
+            url=f"{self.session.hostname}/v1/{self.session.__ext}/models/{model_id}/collections/{collection_id}/scenarios/{scenario_id}/restore"
+            )
+        return response

xplainable_client-1.2.2/xplainable_client/client/_datasets.py

@@ -0,0 +1,51 @@
+from ._client_cog_base import *
+
+import requests
+import pandas as pd
+
+
+class Datasets(Client_Cog):
+
+    # upload_dataset
+
+    def list_datasets(self):
+
+        response = requests.get('https://api.xplainable.io/v1/public-datasets')
+        
+        if response.status_code != 200:
+            raise ValueError(f'Unable to list datasets. Check your connection and try again.')
+        
+        return response.json()
+
+    def _get_url_path(self, name):
+        _base = 'xplainablepublic'
+        _type = 'blob.core.windows.net'
+        _loc = 'asset-repository/datasets'
+
+        return f'https://{_base}.{_type}/{_loc}/{name}/data.csv'
+
+    def load_dataset(self, name):
+
+        datasets = self.list_datasets()
+
+        if name not in datasets:
+            raise ValueError(f'{name} is not available. Run xp.list_datasets() to see available datasets.')
+        
+        try:
+            url = self._get_url_path(name)
+            df = pd.read_csv(url)
+
+            return df
+        
+        except Exception as e:
+            raise ValueError(f'Unable to load dataset {name}. Check your connection and try again.')
+        
+    # update_dataset
+
+    # detele_dataset
+
+    # list_dataset_children
+
+    # preview_dataset
+
+    # list_pub_datasets