Skip to content

Commit 257a89b

Browse files
Merge pull request #117 from le-ander/dev
add missing definition of coef_names when dmat is provided directly
2 parents 8842f6b + bc23a98 commit 257a89b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

batchglm/data.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -235,12 +235,16 @@ def constraint_system_from_star(
235235
if isinstance(dmat, pd.DataFrame):
236236
coef_names = dmat.columns
237237
dmat = dmat.values
238+
else:
239+
coef_names = dmat.design_info.column_names
238240
elif constraints is None:
239241
cmat = None
240242
term_names = None
241243
if isinstance(dmat, pd.DataFrame):
242244
coef_names = dmat.columns
243245
dmat = dmat.values
246+
else:
247+
coef_names = dmat.design_info.column_names
244248
else:
245249
raise ValueError("constraint format %s not recognized" % type(constraints))
246250

0 commit comments

Comments
 (0)