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 node IDs for a specific node type.
docstr_tok: ["Get","the","partition","IDs","of","node","IDs","for","a","specific","node","type","."]
code: def get_partition_ids_from_nids(
      self,
      ids: torch.Tensor,
      node_type: Optional[NodeType] = None,
  ) -> Tensor:
      r"""Get the partition IDs of node IDs for a specific node type."""
      if self.meta["is_hetero"]:
          return self.node_pb[node_type][ids]
      else:
          return self.node_pb[ids]
  
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 3 for calling the LocalGraphStore obj's underlying member method code with example usage: obj.get_partition_ids_from_nids(ids, node_type) and returns: self, self