diff --git a/crowdkit/aggregation/classification/mace.py b/crowdkit/aggregation/classification/mace.py index 0d6c1cb..193518d 100644 --- a/crowdkit/aggregation/classification/mace.py +++ b/crowdkit/aggregation/classification/mace.py @@ -313,7 +313,9 @@ def _initialize(self, n_workers: int, n_labels: int) -> None: self.theta_priors_[:, 0] = self.alpha self.theta_priors_[:, 1] = self.beta - self.strategy_priors_ = np.ones((n_workers, n_labels)) * 10.0 + self.strategy_priors_ = np.multiply( + 10.0, np.ones((n_workers, n_labels)), dtype=np.float64 + ) def _e_step( self, diff --git a/crowdkit/aggregation/image_segmentation/segmentation_rasa.py b/crowdkit/aggregation/image_segmentation/segmentation_rasa.py index 5ea4b0a..8ee83b1 100644 --- a/crowdkit/aggregation/image_segmentation/segmentation_rasa.py +++ b/crowdkit/aggregation/image_segmentation/segmentation_rasa.py @@ -105,7 +105,7 @@ def _aggregate_one(self, segmentations: "pd.Series[Any]") -> npt.NDArray[Any]: for _ in range(self.n_iter): weighted = self._segmentation_weighted(segmentations_np, weights) mv = weighted >= 0.5 - weights = self._calculate_weights(segmentations_np, mv) + weights = self._calculate_weights(segmentations_np, mv) # type: ignore[assignment,unused-ignore] if last_aggregated is not None: delta = weighted - last_aggregated diff --git a/crowdkit/aggregation/utils.py b/crowdkit/aggregation/utils.py index 358c16a..7c706c0 100644 --- a/crowdkit/aggregation/utils.py +++ b/crowdkit/aggregation/utils.py @@ -191,7 +191,7 @@ def converter(series: "pd.Series[Any]") -> "pd.Series[Any]": series.name = name return series - return attr.ib(init=False, converter=converter, on_setattr=attr.setters.convert) # type: ignore[no-any-return] + return attr.ib(init=False, converter=converter, on_setattr=attr.setters.convert) def add_skills_to_data( diff --git a/crowdkit/metrics/data/_classification.py b/crowdkit/metrics/data/_classification.py index 4afece2..b66d15d 100644 --- a/crowdkit/metrics/data/_classification.py +++ b/crowdkit/metrics/data/_classification.py @@ -4,7 +4,7 @@ "alpha_krippendorff", ] -from typing import Any, Callable, Hashable, List, Optional, Tuple, Union, cast +from typing import Any, Callable, Hashable, List, Optional, Union, cast import numpy as np import pandas as pd @@ -266,7 +266,5 @@ def alpha_krippendorff( 0.4444444444444444 """ _check_answers(answers) - data: List[Tuple[Any, Hashable, Hashable]] = answers[ - ["worker", "task", "label"] - ].values.tolist() + data = answers[["worker", "task", "label"]].values.tolist() return float(AnnotationTask(data, distance).alpha())