Nit fix for mypy issue (#87)

google · Nov 8, 2022 · 359ea0f · 359ea0f
1 parent 44ae935
commit 359ea0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/train.py b/scripts/train.py
@@ -210,7 +210,7 @@ def fit(X_train: npt.NDArray[np.bool_],
     X_test = device_put(X_test)
     Y_test = device_put(Y_test)
   N_train, M_train = X_train.shape
-  w = jnp.ones(N_train) / N_train  # type: ignore
+  w = jnp.ones(N_train) / N_train
   YX_train = Y_train[:, None] ^ X_train
   for t in range(iters):
     print('=== %s ===' % (t))

diff --git a/setup.cfg b/setup.cfg
@@ -29,7 +29,7 @@ dev =
     html5lib
     isort
     numpy
-    mypy
+    mypy==0.971
     pytest
     toml
     twine
-Original file line number
+Diff line change
@@ Expand Up / @@ -29,7 +29,7 @@ dev = @@
         html5lib
         isort
         numpy
-        mypy
+        mypy==0.971
         pytest
         toml
         twine
@@ Expand Down @@