install numpy-stubs for mypy

ayshox · Jun 3, 2020 · 026464a · 026464a
1 parent 40f9d48
commit 026464a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -10,6 +10,7 @@ repos:
     rev: master
     hooks:
     -   id: mypy
+        additional_dependencies: ['git+https://github.com/numpy/numpy-stubs']
 -   repo: https://github.com/PyCQA/flake8
     rev: master
     hooks:

diff --git a/rednose/__init__.py b/rednose/__init__.py
@@ -7,9 +7,9 @@
 
 class KalmanFilter:
   name = "<name>"
-  initial_x = None
-  initial_P_diag = None
-  Q = None
+  initial_x = np.zeros((0, 0))
+  initial_P_diag = np.zeros((0, 0))
+  Q = np.zeros((0, 0))
   obs_noise: Dict[int, Any] = {}
 
   def __init__(self, generated_dir):