add check if the float can be converted to integer

responsibleai/responsibleai/rai_insights/rai_insights.py

@@ -568,8 +568,12 @@ def _validate_rai_insights_input_parameters(
 
         # Validate that the target column isn't continuous if the
         # user is running classification scenario
+        # To address error thrown from sklearn here:  # noqa: E501
+        # https://github.com/scikit-learn/scikit-learn/blob/main/sklearn/utils/multiclass.py#L197
+        y_data = train[target_column]
         if (task_type == ModelTask.CLASSIFICATION and
-                pd.api.types.is_float_dtype(train[target_column].dtype)):
+                pd.api.types.is_float_dtype(y_data.dtype) and
+                np.any(y_data != y_data.astype(int))):
             raise UserConfigValidationException(
                 "Target column type must not be continuous "
                 "for classification scenario.")