Skip to content

Commit

Permalink
Merge pull request #67 from ihmeuw-msca/fix_initialize_data
Browse files Browse the repository at this point in the history
fix make data folder, move lines down
  • Loading branch information
AHsu98 authored Feb 5, 2024
2 parents d66f07c + e503552 commit a33796f
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/onemod/actions/data/initialize_results.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,6 @@ def initialize_results(experiment_dir: str, stages: list[str]) -> None:

dataif, settings = get_handle(experiment_dir)

for stage in stages:
stage_init_map[stage](dataif)

# ETL the input data into parquet format.
# More compressible, faster IO, allows for partitioning
raw_input_path = settings.input_path
Expand All @@ -46,6 +43,10 @@ def initialize_results(experiment_dir: str, stages: list[str]) -> None:
# Saves to $experiment_dir/data/data.parquet
dataif.dump_data(data)

for stage in stages:
stage_init_map[stage](dataif)



def _initialize_rover_covsel_results(dataif: DataInterface) -> None:
"""Initialize rover results."""
Expand Down

0 comments on commit a33796f

Please sign in to comment.