def _make_req(self, index, src_ids): """ Make rpc request. """ if len(self._meta_path) > len(self._expand_factor): raise ValueError( "input too many meta_path, and can not decide each hops") sampler = strategy2op(self._strategy[index], "Sampler") req = pywrap.new_nbr_req(self._meta_path[index], sampler, self._expand_factor[index]) pywrap.set_nbr_req(req, src_ids) return req
def _make_req(self, ids): sampler = strategy2op(self._strategy, "NegativeSampler") req = pywrap.new_nbr_req(self._object_type, sampler, self._expand_factor) pywrap.set_nbr_req(req, ids) return req