Added checker for non exist column.

Author: lu-ohai

ads/dataset/dataset_with_target.py

@@ -200,6 +200,11 @@ def from_dataframe(
                 DatasetDefaults.sampling_confidence_interval,
                 **init_kwargs,
             )
+
+        if target not in df:
+            raise ValueError(
+                f"{target} column doesn't exist in data frame. Specify a valid one instead."
+            )
 
         if target_type is None:
             target_type = get_target_type(target, sampled_df, **init_kwargs)

tests/unitary/with_extras/dataset/test_dataset_target.py

@@ -6,6 +6,7 @@
 import os
 from typing import Tuple
 import pandas as pd
+import pytest
 from ads.dataset.classification_dataset import BinaryClassificationDataset
 from ads.dataset.dataset_with_target import ADSDatasetWithTarget
 from ads.dataset.pipeline import TransformerPipeline
@@ -46,6 +47,16 @@ def test_accessor_with_target(self):
         assert isinstance(employees, BinaryClassificationDataset)
         self.assert_dataset(employees)
 
+    def test_accessor_with_target_error(self):
+        df=pd.read_csv(self.get_data_path())
+        wrong_column = "wrong_column"
+        with pytest.raises(
+            ValueError, match=f"{wrong_column} column doesn't exist in data frame. Specify a valid one instead."
+        ):
+            employees = df.ads.dataset_with_target(
+                target=wrong_column
+            )
+
     def assert_dataset(self, dataset):
         assert isinstance(dataset.df, pd.DataFrame)
         assert isinstance(dataset.shape, Tuple)