Skip to content

Commit

Permalink
qmapping may use a different dataset for simulation than ml samples
Browse files Browse the repository at this point in the history
  • Loading branch information
henryaddison committed Mar 25, 2024
1 parent ecfe293 commit 5ab7a40
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/ml_downscaling_emulator/bin/postprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,6 +119,7 @@ def filter(
def qm(
workdir: Path,
checkpoint: str = typer.Option(...),
sim_dataset: str = typer.Option(...),
train_dataset: str = typer.Option(...),
train_input_xfm: str = "stan",
eval_dataset: str = typer.Option(...),
Expand All @@ -128,7 +129,7 @@ def qm(
):
# to compute the mapping, use train split data
# open train split of dataset for the target_pr
sim_train_da = open_raw_dataset_split(train_dataset, "train").sel(
sim_train_da = open_raw_dataset_split(sim_dataset, "train").sel(
ensemble_member=ensemble_member
)["target_pr"]

Expand Down

0 comments on commit 5ab7a40

Please sign in to comment.