fix #25

convoviz/data_analysis.py

@@ -9,7 +9,7 @@
 
 from collections import defaultdict
 from datetime import datetime, timezone
-from typing import TYPE_CHECKING, Unpack
+from typing import TYPE_CHECKING
 
 from matplotlib.figure import Figure
 from nltk import download as nltk_download  # type: ignore[import-untyped]
@@ -21,6 +21,7 @@
 
 if TYPE_CHECKING:
     from PIL.Image import Image
+    from typing_extensions import Unpack
 
     from .utils import GraphKwargs, WordCloudKwargs
 

convoviz/long_runs.py

@@ -3,11 +3,13 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import TYPE_CHECKING, Unpack
+from typing import TYPE_CHECKING
 
 from tqdm import tqdm
 
 if TYPE_CHECKING:
+    from typing_extensions import Unpack
+
     from .models import ConversationSet
     from .utils import GraphKwargs, WordCloudKwargs
 

convoviz/models/_conversation.py

@@ -8,7 +8,7 @@
 from datetime import datetime, timedelta
 from os import utime as os_utime
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, ClassVar, Unpack
+from typing import TYPE_CHECKING, Any, ClassVar
 
 from orjson import loads
 from pydantic import BaseModel
@@ -27,6 +27,7 @@
 
 if TYPE_CHECKING:
     from PIL.Image import Image
+    from typing_extensions import Unpack
 
     from ._message import AuthorRole
 

convoviz/models/_conversation_set.py

@@ -6,7 +6,7 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Unpack
+from typing import TYPE_CHECKING, Any
 
 from orjson import OPT_INDENT_2, dumps, loads
 from pydantic import BaseModel
@@ -22,6 +22,7 @@
 
     from matplotlib.figure import Figure
     from PIL.Image import Image
+    from typing_extensions import Unpack
 
     from convoviz.utils import GraphKwargs, WordCloudKwargs
 

pyproject.toml

@@ -1,27 +1,6 @@
-[tool.ruff]
-select = ["ALL"]
-ignore = [
-    "ANN101", # missing-type-self
-    "ANN102", # missing-type-cls
-    "TD002",  # missing-todo-author
-    "TD003",  # missing-todo-link
-]
-extend-include = ["*.ipynb"]
-
-[tool.ruff.per-file-ignores]
-"convoviz/cli.py" = ["T201"]   # print
-"*.ipynb" = ["T201"]           # print
-"tests/*" = ["S101", "SLF001"] # assert , private-member-access
-
-
-[tool.mypy]
-strict = true
-enable_incomplete_feature = ["Unpack"]
-
-
 [tool.poetry]
 name = "convoviz"
-version = "0.1.6"
+version = "0.1.7"
 description = "Get analytics and visualizations on your ChatGPT data !"
 authors = ["Mohamed Cheikh Sidiya <mohamedcheikhsidiya77@gmail.com>"]
 license = "MIT"
@@ -58,3 +37,24 @@ mypy = "^1.6.1"
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+
+[tool.ruff]
+select = ["ALL"]
+ignore = [
+    "ANN101", # missing-type-self
+    "ANN102", # missing-type-cls
+    "TD002",  # missing-todo-author
+    "TD003",  # missing-todo-link
+]
+extend-include = ["*.ipynb"]
+
+[tool.ruff.per-file-ignores]
+"convoviz/cli.py" = ["T201"]   # print
+"*.ipynb" = ["T201"]           # print
+"tests/*" = ["S101", "SLF001"] # assert , private-member-access
+
+
+[tool.mypy]
+strict = true
+enable_incomplete_feature = ["Unpack"]