Add stubs

.github/workflows/test.yml

@@ -36,7 +36,10 @@ jobs:
         echo "VIRTUAL_ENV=$GITHUB_WORKSPACE/.venv" >> $GITHUB_ENV
     - name: Set up Maturin
       run: |
-        uv pip install pip maturin
+        uv pip install pip maturin mypy
         maturin develop
+    - uses: chartboost/ruff-action@v1
+    - name: Run Mypy
+      run: mypy .
     - name: Run tests (Python)
       run: python3 -m unittest discover python

pyproject.toml

@@ -7,12 +7,24 @@ name = "evalica"
 requires-python = ">= 3.8"
 dependencies = ["numpy~=1.26.4"]
 classifiers = [
+    "Development Status :: 2 - Pre-Alpha",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "License :: OSI Approved :: Apache Software License",
     "Programming Language :: Rust",
     "Programming Language :: Python :: Implementation :: CPython",
     "Programming Language :: Python :: Implementation :: PyPy",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Scientific/Engineering :: Information Analysis",
+    "Typing :: Typed",
 ]
 dynamic = ["version"]
 
 [tool.maturin]
 features = ["pyo3/extension-module"]
 python-source = "python"
+
+[tool.mypy]
+ignore_missing_imports = true
+plugins = ["numpy.typing.mypy_plugin"]
+strict = true

python/evalica/evalica.pyi

@@ -0,0 +1,15 @@
+from typing import Tuple
+
+import numpy as np
+import numpy.typing as npt
+
+def py_bradley_terry(
+    m: npt.NDArray[np.int64]
+) -> Tuple[int, npt.NDArray[np.float64]]: ...
+
+def py_newman(
+    m: npt.NDArray[np.int64],
+    seed: int,
+    tolerance: float,
+    max_iter: int
+) -> Tuple[int, npt.NDArray[np.float64]]: ...

python/evalica/py.typed

@@ -0,0 +1 @@
+partial