Skip to content

Commit a5f40a5

Browse files
committed
test fixes for DC sampler
1 parent ffa795b commit a5f40a5

File tree

1 file changed

+5
-0
lines changed
  • sup3r/preprocessing/samplers

1 file changed

+5
-0
lines changed

sup3r/preprocessing/samplers/dc.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def __init__(
3030
sample_shape: Optional[tuple] = None,
3131
batch_size: int = 16,
3232
feature_sets: Optional[dict] = None,
33+
mode: str = 'lazy',
3334
spatial_weights: Optional[
3435
Union[np.ndarray, da.core.Array, list]
3536
] = None,
@@ -57,6 +58,9 @@ def __init__(
5758
Optional dictionary describing how the full set of features is
5859
split between `lr_only_features` and `hr_exo_features`. See
5960
:class:`~sup3r.preprocessing.Sampler`
61+
mode : str
62+
Loading mode for sampling.
63+
See :class:`~sup3r.preprocessing.Sampler`
6064
spatial_weights : Union[np.ndarray, da.core.Array] | List | None
6165
Set of weights used to initialize the spatial sampling. e.g. If we
6266
want to start off sampling across 2 spatial bins evenly this should
@@ -76,6 +80,7 @@ def __init__(
7680
sample_shape=sample_shape,
7781
batch_size=batch_size,
7882
feature_sets=feature_sets,
83+
mode=mode,
7984
)
8085

8186
def update_weights(self, spatial_weights, temporal_weights):

0 commit comments

Comments
 (0)