FIX: Fix type annotation for mypy

Author: Zach Banks

databento/common/dbnstore.py

@@ -1331,7 +1331,7 @@ def __init__(
         count: int | None = None,
     ) -> None:
         self._reader = reader
-        self._dtype = np.dtype(dtype)
+        self._dtype: np.typing.DTypeLike = np.dtype(dtype)
         self._offset = offset
         self._count = count
         self._close_on_next = False

pyproject.toml

@@ -73,6 +73,7 @@ no_strict_optional = true
 warn_no_return = true
 warn_unused_configs = true
 warn_unused_ignores = true
+plugins = ["numpy.typing.mypy_plugin"]
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]

scripts/lint.sh

@@ -1,4 +1,3 @@
 #!/usr/bin/env bash
-echo $(mypy --version)
-echo Running mypy...
-poetry run mypy databento examples tests
+echo "Running $(poetry run mypy --version)..."
+poetry run mypy .