def _get_call_kwargs(self, ds): attr, collection = ds.get_attr(self._attr) # _call might need to operate on the dedicated instantiated rng # e.g. if seed int is provided return { 'limit_filter': get_limit_filter(self._limit, collection), 'rng': get_rng(self._rng) }
def _get_call_kwargs(self, ds): # determine to be permuted attribute to find the collection pattr = self._pattr if isinstance(pattr, str): pattr, collection = ds.get_attr(pattr) else: # must be sequence of attrs, take first since we only need the shape pattr, collection = ds.get_attr(pattr[0]) # _call might need to operate on the dedicated instantiated rng # e.g. if seed int is provided return { 'limit_filter': get_limit_filter(self._limit, collection), 'rng': get_rng(self.rng) }