File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
sup3r/preprocessing/samplers Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments