Skip to content

Commit cc99d23

Browse files
FIX: Corrected sensitive_attributes comparison in test_superset.py
1 parent d5396ec commit cc99d23

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/metrics/test_superset.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -170,10 +170,10 @@ def test_super_set_fairness_metrics(
170170

171171
for result, expected_result in zip(results, expected_results):
172172
assert isinstance(result, SupersetFairnessRankingResult)
173-
assert isinstance(result.sensitive_attributes, list)
173+
assert isinstance(result.sensitive_attributes, tuple)
174174
assert isinstance(result.rankings, dict)
175175

176-
assert result.sensitive_attributes == expected_result["sensitive_attributes"]
176+
assert result.sensitive_attributes == tuple(expected_result["sensitive_attributes"])
177177

178178
for metric_name, expected_metric_results in expected_result["rankings"].items():
179179
if metric_name in result.rankings:
@@ -296,7 +296,7 @@ def test_super_set_performance_metrics(
296296
assert isinstance(result.sensitive_attributes, tuple)
297297
assert isinstance(result.data, list)
298298

299-
assert result.sensitive_attributes == expected_result["sensitive_attributes"]
299+
assert result.sensitive_attributes == tuple(expected_result["sensitive_attributes"])
300300

301301
result_list = result.data
302302
expected_result_list = expected_result["data"]

0 commit comments

Comments
 (0)