From 9bca9b97a97237e017a6045cc457dd4d33e328fe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Czy=C5=BC?= <pczyz@protonmail.com>
Date: Fri, 26 Apr 2024 13:10:30 +0200
Subject: [PATCH] Silence type errors.

---
 src/bmi/estimators/external/gmm.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/bmi/estimators/external/gmm.py b/src/bmi/estimators/external/gmm.py
index c1cf12f6..e3811ced 100644
--- a/src/bmi/estimators/external/gmm.py
+++ b/src/bmi/estimators/external/gmm.py
@@ -1,7 +1,7 @@
 try:
-    import numpyro
-    import numpyro.distributions as dist
-    from numpyro.infer import MCMC, NUTS
+    import numpyro  # type: ignore
+    import numpyro.distributions as dist  # type: ignore
+    from numpyro.infer import MCMC, NUTS  # type: ignore
 except ImportError:
     numpyro, dist, MCMC, NUTS = [None] * 4
 
@@ -38,7 +38,7 @@ def model(
     """
     alpha = alpha or 1.0 / K
 
-    n_points, n_dim = data.shape
+    _, n_dim = data.shape
 
     # Prior for mixing proportions
     pi = numpyro.sample("pi", dist.Dirichlet(concentration=alpha * jnp.ones(K)))