Closes UNB-2166 by reverting back to using enums

Author: whoseoyster

docs/source/reference/upload.rst

@@ -23,32 +23,13 @@ Upload functions
    UnboxClient.add_dataset
    UnboxClient.add_dataframe
 
-Model types
------------
+Model / Task types
+------------------
 .. autosummary:: 
    :toctree: api/
-   :template: class.rst
 
    ModelType
-   ModelType.fasttext
-   ModelType.sklearn
-   ModelType.pytorch
-   ModelType.tensorflow
-   ModelType.transformers
-   ModelType.keras
-   ModelType.rasa
-   ModelType.custom
-
-Task types
------------
-.. autosummary:: 
-   :toctree: api/
-   :template: class.rst
-   
    TaskType
-   TaskType.TextClassification
-   TaskType.TabularClassification
-   TaskType.TabularRegression
 
 Objects
 -------
@@ -57,6 +38,6 @@ Objects
    :template: class.rst
 
    Model
-   Model.to_dict
    Dataset
+   Model.to_dict
    Dataset.to_dict

unboxapi/__init__.py

@@ -77,7 +77,7 @@ def add_model(
         description: str = None,
         **kwargs,
     ) -> Model:
-        """Uploads a model.
+        """Uploads a model to the Unbox platform.
 
         Parameters
         ----------
@@ -430,7 +430,7 @@ def add_dataset(
         sep: str = ",",
         description: Optional[str] = None,
     ) -> Dataset:
-        r"""Uploads a dataset from a csv.
+        r"""Uploads a dataset to the Unbox platform (from a csv).
 
         Parameters
         ----------
@@ -627,7 +627,7 @@ def add_dataframe(
         tag_column_name: Optional[str] = None,
         language: str = "en",
     ) -> Dataset:
-        """Uploads a dataset from a dataframe.
+        r"""Uploads a dataset to the Unbox platform (from a pandas DataFrame).
 
         Parameters
         ----------

unboxapi/datasets.py

@@ -1,5 +1,5 @@
 class Dataset:
-    """Dataset class."""
+    """An object containing information about a dataset on the Unbox platform."""
 
     def __init__(self, json):
         self._json = json
@@ -20,10 +20,10 @@ def __repr__(self):
         return f"Dataset({self._json})"
 
     def to_dict(self):
-        """Returns object properties as a dict
+        """Returns object properties as a dict.
 
         Returns
         -------
-        Dict with object properties
+        Dict with object properties.
         """
         return self._json