Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
jcblemai committed Apr 19, 2024
1 parent f13b271 commit 854058a
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions flepimop/gempyor_pkg/src/gempyor/initial_conditions.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,8 @@ def get_from_config(self, sim_id: int, modinf) -> np.ndarray:
if method == "SetInitialConditionsFolderDraw":
ic_df = modinf.read_simID(ftype=self.initial_conditions_config["initial_file_type"], sim_id=sim_id)
else:
ic_df = self.path_prefix / read_df(
self.initial_conditions_config["initial_conditions_file"].get(),
ic_df = read_df(
self.path_prefix / self.initial_conditions_config["initial_conditions_file"].get(),
)

y0 = np.zeros((modinf.compartments.compartments.shape[0], modinf.nsubpops))
Expand Down Expand Up @@ -105,8 +105,7 @@ def get_from_config(self, sim_id: int, modinf) -> np.ndarray:
if method == "InitialConditionsFolderDraw":
ic_df = modinf.read_simID(ftype=self.initial_conditions_config["initial_file_type"].get(), sim_id=sim_id)
elif method == "FromFile":
ic_df = self.path_prefix / read_df(
self.initial_conditions_config["initial_conditions_file"].get(),
ic_df = read_df(self.path_prefix / self.initial_conditions_config["initial_conditions_file"].get(),
)

# annoying conversion because sometime the parquet columns get attributed a timezone...
Expand Down

0 comments on commit 854058a

Please sign in to comment.