def wait(self):
        """Returns the stored value when invoked."""
        fn = self.fn
        C_sampled_subgraph = self.future.wait()
        seed_offsets = self.seed_offsets
        fetching_original_edge_ids_is_optional = (
            self.fetching_original_edge_ids_is_optional
        )
        # Ensure there is no memory leak.
        self.fn = self.future = self.seed_offsets = None
        self.fetching_original_edge_ids_is_optional = None
        return fn(
            C_sampled_subgraph,
            seed_offsets,
            fetching_original_edge_ids_is_optional,
        )
Returns the stored value when invoked.
python/dgl/graphbolt/impl/fused_csc_sampling_graph.py