From 75e80bcaacaec26f40b67bfc5ed16e3e0d2b571d Mon Sep 17 00:00:00 2001 From: vinutha karanth Date: Mon, 26 Jun 2023 09:48:50 -0700 Subject: [PATCH 1/2] warn to exp Signed-off-by: vinutha karanth --- responsibleai/responsibleai/rai_insights/rai_insights.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/responsibleai/responsibleai/rai_insights/rai_insights.py b/responsibleai/responsibleai/rai_insights/rai_insights.py index 89f14f220b..6d280ef9b8 100644 --- a/responsibleai/responsibleai/rai_insights/rai_insights.py +++ b/responsibleai/responsibleai/rai_insights/rai_insights.py @@ -591,9 +591,10 @@ def _validate_rai_insights_input_parameters( if np.any(test[feature].isnull()): list_of_feature_having_missing_values.append(feature) if len(list_of_feature_having_missing_values) > 0: - warnings.warn( + raise UserConfigValidationException( f"Features {list_of_feature_having_missing_values} " - "have missing values in test data") + "have missing values in test data. " + "Please check your data.") self._validate_feature_metadata( feature_metadata, train, task_type, model, target_column) From 3e985c969f317c0001e0e47098a49b15672a0e3a Mon Sep 17 00:00:00 2001 From: vinutha karanth Date: Mon, 26 Jun 2023 09:55:10 -0700 Subject: [PATCH 2/2] up test Signed-off-by: vinutha karanth --- .../tests/rai_insights/test_rai_insights_validations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/responsibleai/tests/rai_insights/test_rai_insights_validations.py b/responsibleai/tests/rai_insights/test_rai_insights_validations.py index 8d6d5e96f0..eda7b22578 100644 --- a/responsibleai/tests/rai_insights/test_rai_insights_validations.py +++ b/responsibleai/tests/rai_insights/test_rai_insights_validations.py @@ -68,8 +68,8 @@ def test_missing_data_warnings(self): model = create_complex_classification_pipeline( X_train, y_train, ['Column1', 'Column2'], []) - with pytest.warns( - UserWarning, + with pytest.raises( + UserConfigValidationException, match="['Column1']"): RAIInsights( model=model,