fix typing

Author: voorhs

autointent/_dump_tools/base.py

@@ -15,7 +15,7 @@
 ModuleAttributes: TypeAlias = (
     ModuleSimpleAttributes
     | TagsList
-    | np.ndarray  # type: ignore[type-arg]
+    | np.ndarray
     | Embedder
     | VectorIndex
     | BaseEstimator

autointent/_dump_tools/unit_dumpers.py

@@ -12,7 +12,7 @@
 from peft import PeftModel
 from pydantic import BaseModel
 from sklearn.base import BaseEstimator
-from transformers import (  # type: ignore[attr-defined]
+from transformers import (
     AutoModelForSequenceClassification,
     AutoTokenizer,
     PreTrainedModel,

autointent/context/data_handler/_stratification.py

@@ -13,7 +13,7 @@
 from numpy import typing as npt
 from sklearn.model_selection import train_test_split
 from skmultilearn.model_selection import IterativeStratification
-from transformers import set_seed  # type: ignore[attr-defined]
+from transformers import set_seed
 
 from autointent import Dataset
 from autointent.custom_types import LabelType

autointent/modules/scoring/_bert.py

@@ -9,7 +9,7 @@
 import torch
 from datasets import Dataset, DatasetDict
 from sklearn.model_selection import train_test_split
-from transformers import (  # type: ignore[attr-defined]
+from transformers import (
     AutoModelForSequenceClassification,
     AutoTokenizer,
     DataCollatorWithPadding,

pyproject.toml

@@ -89,12 +89,12 @@ docs = [
     "sphinx-toolbox  (>=4.0.0,<5.0.0)",
     "sphinx-llms-txt (>=0.3.0,<0.4.0)"
 ]
-# dspy = [
-#     "dspy (>=2.6.5,<3.0.0)",
-# ]
-# wandb = [
-#     "wandb (>=0.19.10,<1.0.0)",
-# ]
+dspy = [
+    "dspy (>=2.6.5,<3.0.0)",
+]
+wandb = [
+    "wandb (>=0.19.10,<1.0.0)",
+]
 # codecarbon = [
 #     "codecarbon (>=3.0.2, <3.1.0)",
 #     "pynvml (>=8.0.4, <12.0.0)", # to avoid "attribute nvmlDeviceGetTotalEnergyConsumption not found" error
@@ -105,6 +105,7 @@ fastapi = [
     "pydantic-settings (>=2.0, <3.0)"
 ]
 fastmcp = [
+    "pydantic-settings (>=2.0, <3.0)",
     "fastmcp (>=2.11.3, <3.0)"
 ]
 
@@ -257,3 +258,10 @@ module = [
 ]
 warn_unreachable = false
 
+[[tool.mypy.overrides]]
+module = [
+    "autointent.server.http",
+    "autointent.server.mcp",
+]
+ignore_errors = true
+