Skip to content
Open
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 1 addition & 7 deletions src/weathergen/datasets/batch.py
Original file line number Diff line number Diff line change
Expand Up @@ -472,13 +472,7 @@ def get_num_source_steps(self) -> int:
"""
Get number of input/source steps from smallest of all available streams
"""
# TODO: define explicitly
lens = [
len(stream.source_tokens_cells)
for _, stream in self.target_samples.samples[0].streams_data.items()
]

return min(lens)
return self.target_samples.get_num_steps()

def get_num_target_steps(self) -> int:
"""
Expand Down
Loading