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):
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)
}
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)
}
