@@ -206,11 +206,9 @@ def sampler(self) -> BaseSampler | GroupSampler:
206206 case CombinedWalkModeEnum .SEQUENTIAL :
207207 sampler = SequentialSampleSelector ()
208208 case CombinedWalkModeEnum .RANDOMGROUPED :
209- sampler = RandomGroupSampleSelector (group = self .params . grouping )
209+ sampler = RandomGroupSampleSelector (group = self .grouping )
210210 case CombinedWalkModeEnum .SEQUENTIALGROUPED :
211- sampler = SequentialGroupSampleSelector (
212- group = self .params .grouping
213- )
211+ sampler = SequentialGroupSampleSelector (group = self .grouping )
214212 case _:
215213 # this can never happen, as we are validating this above
216214 pass
@@ -219,11 +217,11 @@ def sampler(self) -> BaseSampler | GroupSampler:
219217 match self .mode_to_enum ():
220218 case CombinedWalkModeEnum .RANDOMGROUPED :
221219 sampler = ExplicitEntitySpaceGroupedGridSampleGenerator (
222- mode = WalkModeEnum .RANDOM , group = self .params . grouping
220+ mode = WalkModeEnum .RANDOM , group = self .grouping
223221 )
224222 case CombinedWalkModeEnum .SEQUENTIALGROUPED :
225223 sampler = ExplicitEntitySpaceGroupedGridSampleGenerator (
226- mode = WalkModeEnum .SEQUENTIAL , group = self .params . grouping
224+ mode = WalkModeEnum .SEQUENTIAL , group = self .grouping
227225 )
228226 case CombinedWalkModeEnum .RANDOM :
229227 sampler = ExplicitEntitySpaceGridSampleGenerator (
0 commit comments