Fix substitution rule

BlueBrain · Mar 8, 2023 · 5488996 · 5488996
1 parent d6dcc09
commit 5488996
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/synthesis_workflow/synthesis.py b/src/synthesis_workflow/synthesis.py
@@ -67,9 +67,10 @@ def apply_substitutions(original_morphs_df, substitution_rules=None):
 
     new_morphs_df = original_morphs_df.copy()
     for mtype_orig, mtypes in substitution_rules.items():
-        df = original_morphs_df[original_morphs_df.mtype.isin(mtypes)].copy()
-        df["mtype"] = mtype_orig
-        new_morphs_df = pd.concat([new_morphs_df, df])
+        for mtype in mtypes:
+            df = original_morphs_df[original_morphs_df.mtype == mtype_orig].copy()
+            df["mtype"] = mtype
+            new_morphs_df = pd.concat([new_morphs_df, df])
     return new_morphs_df