class_docstr: This class implements the :class:`torch_geometric.data.GraphStore`
  interface to act as a local graph store for distributed training.
docstr: Get the partition IDs of edge IDs for a specific edge type.
docstr_tok: ["Get","the","partition","IDs","of","edge","IDs","for","a","specific","edge","type","."]
code: def get_partition_ids_from_eids(
      self, eids: torch.Tensor, edge_type: Optional[EdgeType] = None
  ):
      r"""Get the partition IDs of edge IDs for a specific edge type."""
      if self.meta["is_hetero"]:
          return self.edge_pb[edge_type][eids]
      else:
          return self.edge_pb[eids]
  
imports: import json
  import os.path
  from typing import Any, Dict, List, Optional, Tuple, Union
  import torch
  from torch import Tensor
  from torch_geometric.data import EdgeAttr, GraphStore
  from torch_geometric.typing import EdgeTensorType, EdgeType, NodeType
  from torch_geometric.utils import sort_edge_index
label_desc: Use image node_id 4 for calling the LocalGraphStore obj's underlying member method code with example usage: obj.get_partition_ids_from_eids(eids, edge_type) and returns: self, self