Make CountCorrector an ABC

czbiohub-sf · Nov 25, 2024 · 3a17077 · 3a17077
1 parent 3d0c042
commit 3a17077
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/stats_utils/corrector.py b/stats_utils/corrector.py
@@ -21,7 +21,7 @@
 from stats_utils.constants RBCS_P_UL
 
 
-class CountCorrector:
+class CountCorrector(ABC):
     def __init__(
         self,
         inv_cmatrix: npt.NDArray,
@@ -35,6 +35,13 @@ def __init__(
         self.rbc_ids = rbc_ids
         self.parasite_ids = parasite_ids
 
+    @abstractmethod
+    def get_res_from_counts(self, raw_counts: npt.NDArray, units_ul_out: bool = False
+    ) -> Tuple[float, float]:
+
+    @abstractmethod
+    def calc_parasitemia(self, counts: npt.NDArray) -> float:
+
     def _correct_counts(self, raw_counts: npt.NDArray):
         """
         Correct raw counts using transformation matrix