Add missing type annotations

adonath · Dec 18, 2024 · b8c7a77 · b8c7a77
1 parent 5f173a1
commit b8c7a77
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gmmx/gmm.py b/gmmx/gmm.py
@@ -433,8 +433,8 @@ def from_k_means(
         n_components: int,
         reg_covar: float = 1e-6,
         covariance_type: CovarianceType = CovarianceType.full,
-        **kwargs,
-    ) -> None:
+        **kwargs: dict,
+    ) -> GaussianMixtureModelJax:
         """Init from k-means clustering
 
         Parameters