Use correct typing.

Author: ilan-gold

diffxpy/testing/det.py

@@ -698,7 +698,7 @@ class DifferentialExpressionTestWald(_DifferentialExpressionTestSingle):
     Single wald test per gene.
     """
 
-    model_estim: glm.train.numpy.nb.model_container
+    model_estim: glm.train.base.BaseModelContainer
     sample_description: pd.DataFrame
     coef_loc_totest: np.ndarray
     theta_mle: np.ndarray
@@ -708,7 +708,7 @@ class DifferentialExpressionTestWald(_DifferentialExpressionTestSingle):
 
     def __init__(
             self,
-            model_estim: glm.train.numpy.nb.model_container,
+            model_estim: glm.train.base.BaseModelContainer,
             col_indices: np.ndarray,
             noise_model: str,
             sample_description: pd.DataFrame

diffxpy/testing/det_cont.py

@@ -20,15 +20,15 @@
 
 class _DifferentialExpressionTestCont(_DifferentialExpressionTestSingle):
     _de_test: _DifferentialExpressionTestSingle
-    _model_estim: glm.train.numpy.nb.model_container
+    _model_estim: glm.train.base.BaseModelContainer
     _size_factors: np.ndarray
     _continuous_coords: np.ndarray
     _spline_coefs: list
 
     def __init__(
             self,
             de_test: _DifferentialExpressionTestSingle,
-            model_estim: glm.train.numpy.nb.model_container,
+            model_estim: glm.train.base.BaseModelContainer,
             size_factors: np.ndarray,
             continuous_coords: np.ndarray,
             spline_coefs: list,

diffxpy/testing/det_pair.py

@@ -341,13 +341,13 @@ class DifferentialExpressionTestZTest(_DifferentialExpressionTestPairwiseBase):
     lazy test evaluation.
     """
 
-    model_estim: glm.train.numpy.nb.model_container
+    model_estim: glm.train.base.BaseModelContainer
     theta_mle: np.ndarray
     theta_sd: np.ndarray
 
     def __init__(
             self,
-            model_estim: glm.train.numpy.nb.model_container,
+            model_estim: glm.train.base.BaseModelContainer,
             grouping,
             groups,
             correction_type: str
@@ -528,13 +528,13 @@ class DifferentialExpressionTestZTestLazy(_DifferentialExpressionTestPairwiseLaz
     memory.
     """
 
-    model_estim: glm.train.numpy.nb.model_container
+    model_estim: glm.train.base.BaseModelContainer
     _theta_mle: np.ndarray
     _theta_sd: np.ndarray
 
     def __init__(
             self,
-            model_estim: glm.train.numpy.nb.model_container,
+            model_estim: glm.train.base.BaseModelContainer,
             grouping, groups,
             correction_type="global"
     ):