adjust naming convention for deconvoluted model

theislab · Sep 10, 2023 · 9c23d03 · 9c23d03
1 parent 4137a39
commit 9c23d03
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ncem/tl/fit/backend/design_matrix.py b/ncem/tl/fit/backend/design_matrix.py
@@ -303,6 +303,7 @@ def get_dmats_from_deconvoluted(
         dmat_columns = [
             col if col.startswith(PREFIX_INDEX) else PREFIX_INDEX + x + col for col in dmats[x].design_info.column_names
         ]
+        dmat_columns = [f"{x.split('[')[0]}{x.split('[')[1].split(']')[-1]}" for x in dmat_columns]
         dmats[x] = pd.DataFrame(np.asarray(dmats[x]), index=obs.index, columns=dmat_columns)
     return dmats