def get_size(self):
        """Get the full size of the data.

        Returns
        -------
        int
            data size
        """
        if self.is_node:
            return self._partition_book._num_nodes(self.type_name)
        else:
            return self._partition_book._num_edges(self.type_name)
Get the full size of the data.

        Returns
        -------
        int
            data size
        
python/dgl/distributed/graph_partition_book.py