Skip to content

Commit ea5908f

Browse files
clessigclessig
andauthored
Fixed bug for empty source (ecmwf#800)
Co-authored-by: clessig <christian.lessig@ecwmf.int>
1 parent 8f694ac commit ea5908f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/weathergen/datasets/stream_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def add_empty_target(self, fstep: int) -> None:
127127

128128
self.target_tokens[fstep] += [[torch.tensor([], dtype=torch.int32)]]
129129
self.target_tokens_lens[fstep] += [torch.zeros([self.nhc_target], dtype=torch.int32)]
130-
self.target_coords[fstep] += [[torch.zeros((0, 106)) for _ in range(self.nhc_target)]]
130+
self.target_coords[fstep] += [[torch.zeros((0, 105)) for _ in range(self.nhc_target)]]
131131
self.target_coords_lens[fstep] += [torch.zeros([self.nhc_target], dtype=torch.int32)]
132132
self.target_coords_raw[fstep] += [[torch.tensor([]) for _ in range(self.nhc_target)]]
133133
self.target_times_raw[fstep] += [

0 commit comments

Comments
 (0)