Activate TC rules

Author: Samoed

docs/source/conf.py

@@ -5,21 +5,26 @@
 
 # -- Project information -----------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
+from __future__ import annotations
+
 import os
 import sys
 from importlib.metadata import version
 from pathlib import Path
-
-from sphinx.application import Sphinx
+from typing import TYPE_CHECKING
 
 conf_dir = os.path.dirname(os.path.abspath(__file__))  # noqa: PTH100, PTH120
 
 sys.path.insert(0, conf_dir)
 
+
 from docs_utils.skip_members import skip_member  # noqa: E402
 from docs_utils.tutorials import generate_tutorial_links_for_notebook_creation  # noqa: E402
 from docs_utils.versions_generator import generate_versions_json  # noqa: E402
 
+if TYPE_CHECKING:
+    from sphinx.application import Sphinx
+
 project = "AutoIntent"
 copyright = "2025, DeepPavlov"
 author = "DeepPavlov"

docs/source/docs_utils/notebook.py

@@ -1,11 +1,15 @@
+from __future__ import annotations
+
 import abc
 import re
-from typing import ClassVar, Literal
+from typing import TYPE_CHECKING, ClassVar, Literal
 
-import nbformat
 from jupytext import jupytext
 from pydantic import BaseModel
 
+if TYPE_CHECKING:
+    import nbformat
+
 
 class ReplacePattern(BaseModel, abc.ABC):
     """

docs/source/docs_utils/skip_members.py

@@ -1,7 +1,10 @@
-from typing import Any
+from __future__ import annotations
 
-from sphinx.application import Sphinx
-from sphinx.ext.autodoc import Options
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from sphinx.application import Sphinx
+    from sphinx.ext.autodoc import Options
 
 
 def skip_member(app: Sphinx, what: str, name: str, obj: Any, skip: bool, options: Options) -> bool | None:  # noqa: ANN401, ARG001

docs/source/docs_utils/versions_generator.py

@@ -1,9 +1,14 @@
+from __future__ import annotations
+
 import json
 import logging
 import os
 import re
 import subprocess
-from pathlib import Path
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pathlib import Path
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)

pyproject.toml

@@ -158,6 +158,7 @@ ignore = [
     "RUF001",  # ambiguous unicode character
     "PLC0415",  # import outside top-level
 ]
+future-annotations = true  # For TC rules
 
 [tool.ruff.lint.per-file-ignores]
 "__init__.py" = ["F401", "D104"]
@@ -176,6 +177,12 @@ max-args = 10
 [tool.ruff.lint.pydocstyle]
 convention = "google"
 
+[tool.ruff.lint.flake8-type-checking]
+strict = true
+
+[tool.ruff.lint.flake8-unused-arguments]
+ignore-variadic-names = true
+
 [tool.pytest.ini_options]
 minversion = "8.0"
 testpaths = [

src/autointent/_callbacks/base.py

@@ -1,8 +1,11 @@
 """Base class for reporters (W&B, TensorBoard, etc)."""
+from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from pathlib import Path
 
 
 class OptimizerCallback(ABC):

src/autointent/_callbacks/callback_handler.py

@@ -1,8 +1,12 @@
-from pathlib import Path
-from typing import Any
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
 
 from autointent._callbacks.base import OptimizerCallback
 
+if TYPE_CHECKING:
+    from pathlib import Path
+
 
 class CallbackHandler(OptimizerCallback):
     """Internal class that just calls the list of callbacks in order."""

src/autointent/_callbacks/emissions_tracker.py

@@ -1,12 +1,15 @@
 """Emissions tracking functionality for monitoring energy consumption and carbon emissions."""
+from __future__ import annotations
 
 import json
 import logging
-from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from autointent._callbacks import OptimizerCallback
 
+if TYPE_CHECKING:
+    from pathlib import Path
+
 logger = logging.getLogger(__name__)
 
 

src/autointent/_callbacks/wandb.py

@@ -1,10 +1,14 @@
+from __future__ import annotations
+
 import logging
 import os
-from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from autointent._callbacks.base import OptimizerCallback
 
+if TYPE_CHECKING:
+    from pathlib import Path
+
 logger = logging.getLogger(__name__)
 
 

src/autointent/_dataset/_dataset.py

@@ -1,17 +1,22 @@
 """Defines the Dataset class and related utilities for handling datasets."""
+from __future__ import annotations
 
 import json
 import logging
 from collections import defaultdict
 from functools import cached_property
 from pathlib import Path
-from typing import Any, TypedDict
+from typing import TYPE_CHECKING, Any, TypedDict
 
 from datasets import Dataset as HFDataset
 from datasets import Sequence, get_dataset_config_names, load_dataset
 
-from autointent.custom_types import LabelWithOOS, Split
-from autointent.schemas import Intent, Tag
+from autointent.custom_types import Split
+from autointent.schemas import Tag
+
+if TYPE_CHECKING:
+    from autointent.custom_types import LabelWithOOS
+    from autointent.schemas import Intent
 
 logger = logging.getLogger(__name__)
 
@@ -72,7 +77,7 @@ def n_classes(self) -> int:
         return len(self.intents)
 
     @classmethod
-    def from_dict(cls, mapping: dict[str, Any]) -> "Dataset":
+    def from_dict(cls, mapping: dict[str, Any]) -> Dataset:
         """Creates a dataset from a dictionary mapping.
 
         Args:
@@ -83,7 +88,7 @@ def from_dict(cls, mapping: dict[str, Any]) -> "Dataset":
         return DictReader().read(mapping)
 
     @classmethod
-    def from_json(cls, filepath: str | Path) -> "Dataset":
+    def from_json(cls, filepath: str | Path) -> Dataset:
         """Loads a dataset from a JSON file.
 
         Args:
@@ -96,7 +101,7 @@ def from_json(cls, filepath: str | Path) -> "Dataset":
     @classmethod
     def from_hub(
         cls, repo_name: str, data_split: str = "default", intent_subset_name: str = Split.INTENTS
-    ) -> "Dataset":
+    ) -> Dataset:
         """Loads a dataset from the Hugging Face Hub.
 
         Args:
@@ -113,7 +118,7 @@ def from_hub(
 
         return DictReader().read(mapping)
 
-    def to_multilabel(self) -> "Dataset":
+    def to_multilabel(self) -> Dataset:
         """Converts dataset labels to multilabel format."""
         for split_name, split in self.items():
             self[split_name] = split.map(self._to_multilabel)