class_docstr: The OGB WikiKG2 dataset.
  
  .. seealso:: https://ogb.stanford.edu/docs/linkprop/#ogbl-wikikg2
  
  ---
  name: OGB WikiKG2
  citation:
      author: Hu
      year: 2020
      link: https://arxiv.org/abs/2005.00687
      github: snap-stanford/ogb
  statistics:
      entities: 2500604
      relations: 535
      training: 16109182
      testing: 598543
      validation: 429456
      triples: 17137181
class_docstr_tok: ["The","OGB","WikiKG2","dataset",".","..","seealso",":",":","https",":","\/\/ogb.stanford.edu\/docs\/linkprop\/","#","ogbl-wikikg2","--","-","name",":","OGB","WikiKG2","citation",":","author",":","Hu","year",":","2020","link",":","https",":","\/\/arxiv.org\/abs\/2005.00687","github",":","snap-stanford\/ogb","statistics",":","entities",":","2500604","relations",":","535","training",":","16109182","testing",":","598543","validation",":","429456","triples",":","17137181"]
code: def _compose_mapped_triples(
      self, data_dict: WikiKG2TrainDict | WikiKG2EvalDict
  ) -> numpy.ndarray:  # noqa: D102
      return numpy.stack(
          [data_dict["head"], data_dict["relation"], data_dict["tail"]],
          axis=-1,
      )
  
imports: from __future__ import annotations
  import abc
  import logging
  import pathlib
  import typing
  from typing import ClassVar, Generic, Literal, Optional, Sequence, TypedDict, TypeVar, Union, cast, overload
  import click
  import numpy
  import pandas
  import torch
  from docdata import parse_docdata
  from more_click import verbose_option
  from .base import LazyDataset
  from ..triples import TriplesFactory
  from ..typing import EntityMapping, RelationMapping
label_desc: Use image node_id 3 for calling the OGBWikiKG2 obj's underlying member method code with example usage: obj._compose_mapped_triples(data_dict) and returns: numpy