File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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" ]
You can’t perform that action at this time.
0 commit comments