Fixes bug with text classification full models

Author: whoseoyster

examples/text-classification/sklearn/banking/demo-banking.ipynb

@@ -585,7 +585,8 @@
     "\n",
     "    def predict_proba(self, input_data_df: pd.DataFrame):\n",
     "        \"\"\"Makes predictions with the model. Returns the class probabilities.\"\"\"\n",
-    "        return self.model.predict_proba(input_data_df)\n",
+    "        text_column = input_data_df.columns[0]\n",
+    "        return self.model.predict_proba(input_data_df[text_column])\n",
     "\n",
     "\n",
     "def load_model():\n",
@@ -750,7 +751,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.13"
+   "version": "3.8.10"
   }
  },
  "nbformat": 4,

examples/text-classification/sklearn/sentiment-analysis/sentiment-sklearn.ipynb

@@ -596,7 +596,8 @@
     "\n",
     "    def predict_proba(self, input_data_df: pd.DataFrame):\n",
     "        \"\"\"Makes predictions with the model. Returns the class probabilities.\"\"\"\n",
-    "        return self.model.predict_proba(input_data_df)\n",
+    "        text_column = input_data_df.columns[0]\n",
+    "        return self.model.predict_proba(input_data_df[text_column])\n",
     "\n",
     "\n",
     "def load_model():\n",
@@ -761,7 +762,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.13"
+   "version": "3.8.10"
   }
  },
  "nbformat": 4,

openlayer/validators.py

@@ -315,7 +315,7 @@ def _validate_bundle_resources(self):
                 sample_data = None
                 if "textColumnName" in validation_dataset_config:
                     sample_data = validation_dataset_df[
-                        validation_dataset_config["textColumnName"]
+                        [validation_dataset_config["textColumnName"]]
                     ].head()
 
                 else:

openlayer/version.py

@@ -1,3 +1,3 @@
 # Define the SDK version here so that the interal package can have access to this value.
 # See https://stackoverflow.com/questions/2058802
-__version__ = "0.0.0a8"
+__version__ = "0.0.0a9"