From 5bcbf991e8af0357614af05c683df94857e9f5ff Mon Sep 17 00:00:00 2001
From: Jianqi Pan <jannchie@gmail.com>
Date: Mon, 29 Jul 2024 23:54:24 +0900
Subject: [PATCH] :wrench: chore(mypy): remove stub

---
 pyproject.toml              |  3 --
 stubs/wdtagger/__init__.pyi | 61 -------------------------------------
 2 files changed, 64 deletions(-)
 delete mode 100644 stubs/wdtagger/__init__.pyi

diff --git a/pyproject.toml b/pyproject.toml
index a256d0f..ac4d07c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,9 +31,6 @@ line-length = 120
 [tool.mypy]
 ignore_missing_imports = true
 
-[tool.poetry.package.include]
-"stubs/**/*.pyi" = {}
-
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
diff --git a/stubs/wdtagger/__init__.pyi b/stubs/wdtagger/__init__.pyi
deleted file mode 100644
index d7e913a..0000000
--- a/stubs/wdtagger/__init__.pyi
+++ /dev/null
@@ -1,61 +0,0 @@
-from pathlib import Path
-from typing import Any, Sequence, Union
-
-import numpy as np
-from _typeshed import Incomplete
-from PIL import Image
-
-Input = Union[np.ndarray, Image.Image, str, Path]
-
-__all__ = ["Tagger"]
-
-class Result:
-    general_tag_data: Incomplete
-    character_tag_data: Incomplete
-    rating_data: Incomplete
-    def __init__(self, pred, sep_tags, general_threshold: float = 0.35, character_threshold: float = 0.9) -> None: ...
-    @property
-    def general_tags(self): ...
-    @property
-    def character_tags(self): ...
-    @property
-    def rating(self): ...
-    @property
-    def general_tags_string(self) -> str: ...
-    @property
-    def character_tags_string(self) -> str: ...
-    @property
-    def all_tags(self) -> list[str]: ...
-    @property
-    def all_tags_string(self) -> str: ...
-
-class Tagger:
-    console: Incomplete
-    logger: Incomplete
-    model_target_size: Incomplete
-    cache_dir: Incomplete
-    hf_token: Incomplete
-    def __init__(
-        self,
-        model_repo: str = "SmilingWolf/wd-swinv2-tagger-v3",
-        cache_dir: Incomplete | None = None,
-        hf_token=...,
-        loglevel=...,
-        num_threads: Incomplete | None = None,
-        providers: Incomplete | None = None,
-        console: Incomplete | None = None,
-    ) -> None: ...
-    sep_tags: Incomplete
-    model: Incomplete
-    def load_model(
-        self,
-        model_repo,
-        cache_dir: Incomplete | None = None,
-        hf_token: Incomplete | None = None,
-        num_threads: int | None = None,
-        providers: Sequence[str | tuple[str, dict[Any, Any]]] | None = None,
-    ): ...
-    def pil_to_cv2_numpy(self, image): ...
-    def tag(
-        self, image: Input | list[Input], general_threshold: float = 0.35, character_threshold: float = 0.9
-    ) -> Result | list[Result]: ...