fix: mypy error

shenxiangzhuang · shenxiangzhuang · commit 9b1d29942dcc · 2025-01-14T11:53:03.000+08:00
diff --git a/tests/aggregation/test_ds_aggregation.py b/tests/aggregation/test_ds_aggregation.py
@@ -405,7 +405,7 @@ def test_aggregate_hds_on_toy_ysda(
     )
 
     assert_series_equal(
-        OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(toy_answers_df).sort_index(),  # type: ignore
+        OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(toy_answers_df).sort_index(),
         toy_ground_truth_df.sort_index(),
     )
 
@@ -441,7 +441,7 @@ def test_aggregate_hds_on_simple(
     )
 
     assert_series_equal(
-        OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(simple_answers_df).sort_index(),  # type: ignore
+        OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(simple_answers_df).sort_index(),
         simple_ground_truth.sort_index(),
     )
 

Original file line number	Diff line number	Diff line change
`@@ -405,7 +405,7 @@ def test_aggregate_hds_on_toy_ysda(`
`405`	`405`	`)`
`406`	`406`
`407`	`407`	`assert_series_equal(`
`408`		`- OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(toy_answers_df).sort_index(), # type: ignore`
	`408`	`+ OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(toy_answers_df).sort_index(),`
`409`	`409`	`toy_ground_truth_df.sort_index(),`
`410`	`410`	`)`
`411`	`411`
`@@ -441,7 +441,7 @@ def test_aggregate_hds_on_simple(`
`441`	`441`	`)`
`442`	`442`
`443`	`443`	`assert_series_equal(`
`444`		`- OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(simple_answers_df).sort_index(), # type: ignore`
	`444`	`+ OneCoinDawidSkene(n_iter=n_iter, tol=tol).fit_predict(simple_answers_df).sort_index(),`
`445`	`445`	`simple_ground_truth.sort_index(),`
`446`	`446`	`)`
`447`	`447`