Skip to content

Commit 09eb4c5

Browse files
authored
Hot fix for error in validation_io when source is empty but target not (#44)
1 parent 2ab5208 commit 09eb4c5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/weathergen/utils/validation_io.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def write_validation(
106106
else:
107107
write_first = True
108108

109-
# TODO: how to avoid this
109+
# TODO: how to avoid the case distinction for write_first
110110
if write_first:
111111
ds_source = ds.require_group(f"{rn}/{fs}")
112112
# column names
@@ -127,7 +127,8 @@ def write_validation(
127127
ds_source.create_dataset("targets_lens", data=targets_lens_k)
128128
else:
129129
rn = rn + f"/{fs}"
130-
ds[f"{rn}/sources"].append(source_k)
130+
if source_lens_k.sum() > 0 :
131+
ds[f"{rn}/sources"].append(source_k)
131132
ds[f"{rn}/sources_lens"].append(source_lens_k)
132133
ds[f"{rn}/preds"].append(preds_k)
133134
ds[f"{rn}/targets"].append(targets_k)

0 commit comments

Comments
 (0)