class_docstr: A DataSet to lazily load graphs based on queries from TypeDB and some arbitrary id.
code: def __init__(
      self,
      indices: Sequence,
      node_types,
      edge_type_triplets,
      queries_for_id: Callable,
      session: Optional[TypeDBSession] = None,
      infer: bool = True,
      transform: Optional[Callable[[nx.Graph], nx.Graph]] = None,
  ):
      self._indices = indices
      self._node_types = node_types
      self._edge_type_triplets = edge_type_triplets
      self.queries_for_id = queries_for_id
      self._infer = infer
      self._transform = transform
      self.session = session
  
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 1 to create a new DataSet object  with example: obj = DataSet(indices, node_types, edge_type_triplets, queries_for_id, session, infer, transform)