Switch mypy to ty

Author: F1uctus

flake.lock

@@ -1,59 +1,43 @@
-[tool.poetry]
+[project]
 name = "ttc"
 version = "0.1.0"
 description = "A Text-To-Conversation Natural Language Processing toolkit."
-authors = ["Ilya I. Nikitin <ilya.i.nikitin@proton.me>"]
-license = "GPL-3.0+"
 readme = "README.md"
-repository = "https://github.com/f1uctus/ttc"
-
-[tool.poetry.dependencies]
-# required
-python = ">=3.8,<3.13"
-click = ">=8.0,<9.0"
-numpy = ">=1.24.0,<2.0.0"
-spacy = ">=3.5.0,<4.0.0"
-ru_core_news_sm = { url = "https://github.com/explosion/spacy-models/releases/download/ru_core_news_sm-3.7.0/ru_core_news_sm-3.7.0.tar.gz" }
-
-[tool.poetry.group.large_models_ru]
-optional = true
-
-[tool.poetry.group.large_models_ru.dependencies]
-ru_core_news_md = { url = "https://github.com/explosion/spacy-models/releases/download/ru_core_news_md-3.7.0/ru_core_news_md-3.7.0.tar.gz" }
-ru_core_news_lg = { url = "https://github.com/explosion/spacy-models/releases/download/ru_core_news_lg-3.7.0/ru_core_news_lg-3.7.0.tar.gz" }
-
-[tool.poetry.group.dev]
-optional = true
-
-[tool.poetry.group.dev.dependencies]
-pytest = { version = ">=7.2.1" }
-jupyter = { version = ">=1.0.0" }
-mypy = { version = ">=1.0" }
-black = { version = ">=23.1" }
-ruff = {version = ">=0.0.275"}
-
-[tool.poetry.scripts]
-ttc = "ttc:cli.cli"
+license = "GPL-3.0-or-later"
+authors = [
+  { name = "Ilya Nikitin", email = "ilya.i.nikitin@proton.me" }
+]
+requires-python = ">=3.10"
+dependencies = [
+  "click>=8.0,<9.0",
+  "numpy>=1.24.0,<2.0.0",
+  "spacy>=3.8,<4.0",
+  "ru_core_news_sm @ https://github.com/explosion/spacy-models/releases/download/ru_core_news_sm-3.8.0/ru_core_news_sm-3.8.0.tar.gz"
+]
+
+[tool.uv.build-backend]
+module-name = "ttc"
+module-root = "."
 
 [build-system]
-requires = ["poetry-core>=1.0.0"]
-build-backend = "poetry.core.masonry.api"
-
-[tool.mypy]
-python_version = "3.8"
-implicit_reexport = true
-show_error_context = true
-show_column_numbers = true
-show_error_codes = true
-warn_return_any = true
-warn_unused_configs = true
-warn_unused_ignores = true
-warn_redundant_casts = true
-strict_optional = true
-
-[[tool.mypy.overrides]]
-module = "tests.*"
-ignore_errors = true
+requires = ["uv_build"]
+build-backend = "uv_build"
+
+[project.optional-dependencies]
+large_models_ru = [
+  "ru_core_news_md @ https://github.com/explosion/spacy-models/releases/download/ru_core_news_md-3.8.0/ru_core_news_md-3.8.0.tar.gz",
+  "ru_core_news_lg @ https://github.com/explosion/spacy-models/releases/download/ru_core_news_lg-3.8.0/ru_core_news_lg-3.8.0.tar.gz"
+]
+dev = [
+  "pytest>=7.2.1",
+  "jupyter>=1.0.0",
+  "ty",
+  "black>=23.1",
+  "ruff>=0.0.275"
+]
+
+[project.scripts]
+ttc = "ttc.cli:cli"
 
 [tool.black]
 line-length = 88

pyproject.toml

@@ -4,7 +4,7 @@
 
 import click
 from click import echo
-from colorama import init as colorama_init, Fore, Back, Style  # type: ignore
+from colorama import init as colorama_init, Fore, Back, Style
 from spacy.tokens import Span
 
 import ttc
@@ -35,6 +35,8 @@ def print_play(file: TextIO, language, with_text: bool):
         echo("Specified language is not supported")
         exit(1)
 
+    assert cc is not None
+
     text = file.read().split("-" * 20)[0]
     file.close()
 
@@ -51,7 +53,7 @@ def print_play(file: TextIO, language, with_text: bool):
     }
 
     echo("Actors found:")
-    echo(", ".join(c + s.text + Style.RESET_ALL for _, (s, c) in actor_colors.items()))
+    echo(", ".join(c + s.text + str(Style.RESET_ALL) for _, (s, c) in actor_colors.items()))
 
     first_col_w = max(len(str(s)) for s in play.actors)
     for r, s in play.lines:

ttc/cli.py

@@ -1,4 +1,5 @@
 import itertools
+from collections import OrderedDict
 from typing import (
     Iterable,
     Iterator,
@@ -7,7 +8,6 @@
     List,
     Tuple,
     Dict,
-    OrderedDict,
 )
 
 T = TypeVar("T")

ttc/iterables.py

@@ -81,7 +81,7 @@ def trim(self: Span, should_trim: Callable[[Token], bool]) -> Span:
 
 @span_extension("method")
 def trim_non_word(self: Span) -> Span:
-    return trim(self, non_word)  # type: ignore
+    return trim(self, non_word)
 
 
 @span_extension("method")

ttc/language/common/span_extensions.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Dict
+from typing import Dict, Any
 
 import spacy
 from spacy import Language
@@ -32,7 +32,7 @@ def __init__(self):
 
         russian_pipelines.register_for(self.language)
 
-        self.language.vocab.get_noun_chunks = noun_chunks
+        self.language.vocab.get_noun_chunks = noun_chunks  # type: ignore
 
         if not Doc.has_extension("nl_indices"):
             Doc.set_extension("nl_indices", default=frozenset())

ttc/language/russian/conversation_classifier.py

@@ -61,7 +61,7 @@ def flush_replica(*tags: Callable[[Span], Any]):
         if tokens:
             replica_span = doc[tokens[0].i : tokens[-1].i + 1]
             for tag in tags:
-                replica_span._.set(tag.__name__, True)
+                replica_span._.set(getattr(tag, "__name__", str(tag)), True)
             replicas.append(replica_span)
             tokens.clear()
 
@@ -177,7 +177,7 @@ def flush_replica(*tags: Callable[[Span], Any]):
                     doc[pt.i : line_end_i + 1], as_spans=True
                 )
                 for match in results:
-                    match: Span = match  # type: ignore
+                    match: Span = match
                     # may be an interrogative or exclamatory ending of a speech
                     # e.g. После всего этого, — что ты еще сказал?
                     if match[0] != pt and match[0].is_punct: