From 70da8f0bb7f67746c95b29c9c4b3a70e46db62a9 Mon Sep 17 00:00:00 2001 From: Responsible ML Infra Date: Wed, 16 Aug 2023 14:26:47 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 557602624 --- .../tutorial_utils/util_test.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/fairness_indicators/tutorial_utils/util_test.py b/fairness_indicators/tutorial_utils/util_test.py index 2068b5da..a72a18b4 100644 --- a/fairness_indicators/tutorial_utils/util_test.py +++ b/fairness_indicators/tutorial_utils/util_test.py @@ -254,21 +254,21 @@ def test_convert_data_csv(self): df = pd.read_csv(output_file).replace("'", '', regex=True) expected_df = pd.DataFrame() - expected_df = expected_df.append( + expected_df = pd.concat([expected_df, pd.DataFrame.from_dict( { 'comment_text': - 'comment 1', + ['comment 1'], 'toxicity': - 0.0, - 'gender': [], - 'sexual_orientation': ['bisexual'], - 'race': ['other_race_or_ethnicity'], - 'religion': ['atheist', 'other_religion'], - 'disability': [ + [0.0], + 'gender': [[]], + 'sexual_orientation': [['bisexual']], + 'race': [['other_race_or_ethnicity']], + 'religion': [['atheist', 'other_religion']], + 'disability': [[ 'physical_disability', 'intellectual_or_learning_disability', 'psychiatric_or_mental_illness', 'other_disability' - ] - }, + ]] + })], ignore_index=True) self.assertEqual(