try to fix mypy

voorhs · voorhs · commit f9ca4c4e05ac · 2025-06-10T20:06:52.000+03:00
diff --git a/autointent/_dump_tools.py b/autointent/_dump_tools.py
@@ -26,7 +26,7 @@
 ModuleSimpleAttributes = None | str | int | float | bool | list  # type: ignore[type-arg]
 
 ModuleAttributes: TypeAlias = (
-    ModuleSimpleAttributes | TagsList | np.ndarray | Embedder | VectorIndex | BaseEstimator | Ranker  # type: ignore[type-arg]
+    ModuleSimpleAttributes | TagsList | np.ndarray | Embedder | VectorIndex | BaseEstimator | Ranker
 )
 
 logger = logging.getLogger(__name__)
diff --git a/autointent/metrics/retrieval.py b/autointent/metrics/retrieval.py
@@ -539,7 +539,7 @@ def retrieval_ndcg(query_labels: LABELS_VALUE_TYPE, candidates_labels: CANDIDATE
     query_label_, candidates_labels_ = transform(query_labels, candidates_labels)
 
     ndcg_scores: list[float] = []
-    relevance_scores: npt.NDArray[np.bool] = query_label_[:, None] == candidates_labels_
+    relevance_scores = query_label_[:, None] == candidates_labels_
 
     for rel_scores in relevance_scores:
         cur_dcg = _dcg(rel_scores, k)

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@`
`26`	`26`	`ModuleSimpleAttributes = None \| str \| int \| float \| bool \| list # type: ignore[type-arg]`
`27`	`27`
`28`	`28`	`ModuleAttributes: TypeAlias = (`
`29`		`- ModuleSimpleAttributes \| TagsList \| np.ndarray \| Embedder \| VectorIndex \| BaseEstimator \| Ranker # type: ignore[type-arg]`
	`29`	`+ ModuleSimpleAttributes \| TagsList \| np.ndarray \| Embedder \| VectorIndex \| BaseEstimator \| Ranker`
`30`	`30`	`)`
`31`	`31`
`32`	`32`	`logger = logging.getLogger(__name__)`