fix: use Union type hint for py3.9 compat

davidlougheed · Jun 10, 2024 · 53e4260 · 53e4260
1 parent 065789b
commit 53e4260
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/strkit/call/allele.py b/strkit/call/allele.py
@@ -261,7 +261,7 @@ def call_alleles(
 
     gmm_cache = {}
 
-    def _get_fitted_gmm(s: NDArray[np.int_] | NDArray[np.float_]) -> Optional[object]:
+    def _get_fitted_gmm(s: Union[NDArray[np.int_], NDArray[np.float_]]) -> Optional[object]:
         if (s_t := tuple(s)) 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)

diff --git a/strkit/mi/expansionhunter.py b/strkit/mi/expansionhunter.py
@@ -12,7 +12,7 @@
 __all__ = ["ExpansionHunterCalculator"]
 
 
-def _parse_allele(a: int | str | None) -> int | None:
+def _parse_allele(a: Union[int, str, None]) -> int | None:
     if isinstance(a, str):
         if a == ".":
             return None