simplify iteration over dataloader within stream

grassesi · grassesi · commit 8ab11e809358 · 2025-10-15T09:13:36.000+02:00
diff --git a/src/weathergen/datasets/multi_stream_data_sampler.py b/src/weathergen/datasets/multi_stream_data_sampler.py
@@ -255,7 +255,7 @@ def __iter__(self):
                     )
 
                     # for all sources for current stream
-                    for _, ds in enumerate(stream_ds):
+                    for ds in stream_ds:
                         # source window (of potentially multi-step length)
                         rdata: ReaderData = ds.get_source(idx)
 

Original file line number	Diff line number	Diff line change
`@@ -255,7 +255,7 @@ def __iter__(self):`
`255`	`255`	`)`
`256`	`256`
`257`	`257`	`# for all sources for current stream`
`258`		`- for _, ds in enumerate(stream_ds):`
	`258`	`+ for ds in stream_ds:`
`259`	`259`	`# source window (of potentially multi-step length)`
`260`	`260`	`rdata: ReaderData = ds.get_source(idx)`
`261`	`261`