Class RandomChoiceGenerator

dask/array/_array_expr/random.py:1030–1052

class RandomChoiceGenerator(RandomChoice):
    _defaults = {}

    @cached_property
    def state_data(self):
        return _spawn_bitgens(self._state, len(self.sizes))

    def _layer(self) -> dict:
        return {
            k: (
                _choice_rng,
                bitgen,
                self.array,
                size,
                self.replace,
                self.p,
                self.axis,
                self.shuffle,
            )
            for k, bitgen, size in zip(
                self.__dask_keys__(), self.state_data, self.sizes
            )
        }


"""