perf(call): use bytes as gmm cache key

davidlougheed · Jun 15, 2024 · 5bf3e14 · 5bf3e14
1 parent e1a401e
commit 5bf3e14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/strkit/call/allele.py b/strkit/call/allele.py
@@ -266,7 +266,7 @@ def call_alleles(
     gmm_cache = {}
 
     def _get_fitted_gmm(s: Union[NDArray[np.int_], NDArray[np.float_]]) -> Optional[object]:
-        if (s_t := tuple(s)) not in gmm_cache:
+        if (s_t := s.tobytes()) not in gmm_cache:
             # Fit Gaussian mixture model to the resampled data
             gmm_cache[s_t] = fit_gmm(rng, s, n_alleles, allele_filter, params.hq, gm_filter_factor)