From b8c7a77b9843a457969f56dd07f8b8329804879f Mon Sep 17 00:00:00 2001
From: Axel Donath <axel.donath@cfa.harvard.edu>
Date: Wed, 18 Dec 2024 13:44:32 -0500
Subject: [PATCH] Add missing type annotations

---
 gmmx/gmm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gmmx/gmm.py b/gmmx/gmm.py
index 83a4502..fb9d511 100644
--- 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