second commit

.gitignore

@@ -0,0 +1,10 @@
+.DS_Store
+.idea
+*.log
+tmp/
+
+*.py[cod]
+*.egg
+build
+htmlcov
+__pycache__

Makefile

@@ -0,0 +1,5 @@
+test:
+	$(MAKE) mypy
+
+mypy:
+	poetry run mypy examples

examples/example.py

@@ -14,19 +14,13 @@
 #         )
 #     ]
 # )
+from naivesearch import InvertedIndex
+
 
 def split_sentence(s):
     return list(s)
 
 
-class InvertedIndex:
-    def __init__(self, path_to_file, chunkers):
-        pass
-
-    def __getitem__(self, q):
-        return ['hoge', 'hoo']
-
-
 index = InvertedIndex(
     '/path/to/file', # FileReader('/path/to/file')
     [

mypy.ini

@@ -0,0 +1,8 @@
+[mypy]
+check_untyped_defs = True
+ignore_errors = False
+ignore_missing_imports = True
+strict_optional = True
+warn_unused_ignores = True
+warn_redundant_casts = True
+warn_unused_configs = True

naivesearch/__init__.py

@@ -0,0 +1,15 @@
+from typing import List, Optional, Protocol
+
+class Chunker(Protocol):
+    def __call__(self, inputs: str) -> List[str]:
+        ...
+
+
+class InvertedIndex:
+    def __init__(self, path_to_file: str, chunkers: List[Chunker]):
+        pass
+
+    def __getitem__(self, q):
+        return ['hoge', 'hoo']
+
+

pyproject.toml

@@ -8,6 +8,7 @@ authors = ["Your Name <[email protected]>"]
 python = "3.8.6"
 
 [tool.poetry.dev-dependencies]
+mypy = "^0.790"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]