@@ -358,10 +358,10 @@ def __init__(
358358 self .groups = list (np .asarray (groups ))
359359
360360 # Values of parameter estimates: coefficients x genes array with one coefficient per group
361- self ._theta_mle = model_estim .a_var
361+ self ._theta_mle = model_estim .model_container . theta_location
362362 # Standard deviation of estimates: coefficients x genes array with one coefficient per group
363363 # Need .copy() here as nextafter needs mutabls copy.
364- theta_sd = np .diagonal (model_estim .fisher_inv , axis1 = - 2 , axis2 = - 1 ).T .copy ()
364+ theta_sd = np .diagonal (model_estim .model_container . fisher_inv , axis1 = - 2 , axis2 = - 1 ).T .copy ()
365365 theta_sd = np .nextafter (0 , np .inf , out = theta_sd , where = theta_sd < np .nextafter (0 , np .inf ))
366366 self ._theta_sd = np .sqrt (theta_sd )
367367 self ._logfc = None
@@ -550,10 +550,10 @@ def __init__(
550550 self .groups = groups .tolist ()
551551
552552 # Values of parameter estimates: coefficients x genes array with one coefficient per group
553- self ._theta_mle = model_estim .a_var
553+ self ._theta_mle = model_estim .model_container . theta_location
554554 # Standard deviation of estimates: coefficients x genes array with one coefficient per group
555555 # Need .copy() here as nextafter needs mutabls copy.
556- theta_sd = np .diagonal (model_estim .fisher_inv , axis1 = - 2 , axis2 = - 1 ).T .copy ()
556+ theta_sd = np .diagonal (model_estim .model_container . fisher_inv , axis1 = - 2 , axis2 = - 1 ).T .copy ()
557557 theta_sd = np .nextafter (0 , np .inf , out = theta_sd , where = theta_sd < np .nextafter (0 , np .inf ))
558558 self ._theta_sd = np .sqrt (theta_sd )
559559
0 commit comments