remove warning

HopkinsIDD · Feb 10, 2024 · 0b77b97 · 0b77b97
1 parent dbad734
commit 0b77b97
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flepimop/gempyor_pkg/src/gempyor/seeding_ic.py b/flepimop/gempyor_pkg/src/gempyor/seeding_ic.py
@@ -489,6 +489,8 @@ def draw(self, sim_id: int, setup) -> np.ndarray:
                         if "rest" in str(ic_df_compartment_val).strip().lower():
                             rests.append([comp_idx, pl_idx])
                         else:
+                            if isinstance(ic_df_compartment_val, pd.Series): # it can also be float if we allow allow_missing_compartments
+                                ic_df_compartment_val = float(ic_df_compartment_val.iloc[0])
                             y0[comp_idx, pl_idx] = float(ic_df_compartment_val)
                 elif allow_missing_subpops:
                     logger.critical(