Merge pull request #204 from le-ander/dev

fix extraction of corf_loc_names when dmat is provided directly
theislab · Jul 15, 2021 · 80478b0 · 80478b0
2 parents 974b482 + 0edf9f8
commit 80478b0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/diffxpy/testing/tests.py b/diffxpy/testing/tests.py
@@ -698,7 +698,10 @@ def wald(
                 as_numeric=as_numeric
             )
         else:
-            coef_loc_names = dmat_loc.columns.tolist()
+            if isinstance(dmat_loc, patsy.design_info.DesignMatrix):
+                coef_loc_names = dmat_loc.design_info.column_names
+            else:
+                coef_loc_names = dmat_loc.columns.tolist()
         if not np.all([x in coef_loc_names for x in coef_to_test]):
             raise ValueError(
                 "the requested test coefficients %s were found in model coefficients %s" %