class_docstr: A DataSet to lazily load graphs based on queries from TypeDB and some arbitrary id.
code: def node_type_indices(self, graph):
      indices = []
      for _, data in graph.nodes(data=True):
          indices.append(self._node_types.index(data["type"]))
      return indices
  
file_path: typedb-ml/typedb_ml/pytorch_geometric/dataset/dataset.py
imports: from typing import Sequence, Callable, Optional
  import networkx
  from torch_geometric.utils import from_networkx
  from typedb.client import TypeDBSession, TypeDBOptions, TransactionType
  from typedb_ml.networkx.queries_to_networkx import build_graph_from_queries
label_desc: Use image node_id 4 for calling the DataSet obj's underlying member method code with example usage: obj.node_type_indices(graph) and returns: indices