Skip to content

Commit 4387b33

Browse files
fix: accessing non-existent field (#137)
1 parent 6a09275 commit 4387b33

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

orchestrator/modules/operators/randomwalk.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)