Client side changes to support tasktype being a property of projects now

Author: Parthib

examples/tabular-classification/churn-classifier/churn-classifier-sklearn.ipynb

@@ -233,20 +233,21 @@
  "metadata": {},
  "outputs": [],
  "source": [
+ "\n",
  "from unboxapi.tasks import TaskType\n",
  "\n",
  "# Comment this out and uncomment the next section to load the project\n",
  "project = client.create_project(\n",
  " name=\"Banking Churn Project\",\n",
- " description=\"Project for predicting Banking Churn\"\n",
+ " description=\"Project for Predicting Banking Churn\",\n",
+ " task_type=TaskType.TabularClassification,\n",
  ")\n",
  "'''\n",
  "# Use this for loading the project on subsequent runs\n",
  "project = client.load_project(\n",
  " name=\"Banking Churn Project\"\n",
  ")\n",
  "'''\n",
- "\n",
  "dataset = project.add_dataframe(\n",
  " df=x_val,\n",
  " class_names=class_names,\n",
@@ -292,6 +293,14 @@
  "metadata": {},
  "outputs": [],
  "source": []
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "id": "b4ecb0cf",
+ "metadata": {},
+ "outputs": [],
+ "source": []
  }
  ],
  "metadata": {

unboxapi/__init__.py

@@ -11,7 +11,7 @@
 from .api import Api
 from bentoml.saved_bundle.bundler import _write_bento_content_to_dir
 from bentoml.utils.tempdir import TempDirectory
-from .exceptions import UnboxException, UnboxInvalidRequest
+from .exceptions import UnboxDuplicateTask, UnboxException, UnboxInvalidRequest
 from .models import Model, ModelType, create_template_model
 from .datasets import Dataset
 from .tasks import TaskType
@@ -62,17 +62,29 @@ def __init__(self, api_key: str):
  else:
  self.upload = self.api.transfer_blob
 
- def create_project(self, name: str, description: str):
+ def create_project(self, name: str, description: str, task_type: TaskType):
  endpoint = "projects"
- payload = dict(name=name, description=description)
+ payload = dict(
+ name=name, 
+ description=description,
+ taskType=task_type.value,
+ )
  project_data = self.api.post_request(endpoint, body=payload)
+ print(project_data)
  return Project(project_data, self.upload, self.subscription_plan, self)
 
  def load_project(self, name: str):
  endpoint = f"me/projects/{name}"
  project_data = self.api.get_request(endpoint)
+ print(project_data)
  return Project(project_data, self.upload, self.subscription_plan, self)
 
+ def create_or_load_project(self, name: str, description: str, task_type: TaskType):
+ try:
+ return self.create_project(name, description, task_type)
+ except UnboxDuplicateTask:
+ return self.load_project(name)
+
  def add_model(
  self,
  name: str,
@@ -440,7 +452,6 @@ def add_model(
  project_id=project_id,
  description=description,
  classNames=class_names,
- taskType=task_type.value,
  architectureType=model_type.name,
  kwargs=list(kwargs.keys()),
  featureNames=feature_names,

unboxapi/api.py

@@ -12,7 +12,7 @@
 
 # UNBOX_ENDPOINT = "https://api-dev.unbox.ai/api"
 # UNBOX_ENDPOINT = "https://api.unbox.ai/api"
-UNBOX_ENDPOINT = "http://localhost:8080/v1"
+UNBOX_ENDPOINT = "http://localhost:8080/api"
 UNBOX_STORAGE_PATH = os.path.expanduser("~/unbox/unbox-onpremise/userStorage")
 
 

unboxapi/projects.py

@@ -41,6 +41,7 @@ def add_model(
  **kwargs,
  ) -> Model:
  kwargs["project_id"] = self.id
+ kwargs["task_type"] = self.taskType # Not camelcase due to API response stored via __getattr__
  return self.client.add_model(*args, **kwargs)
 
  def add_dataset(