code: def load_dataset(root: str, name: str, *args, **kwargs) -> Dataset:
      r"""Returns a variety of datasets according to :obj:`name`."""
      if "karate" in name.lower():
          from torch_geometric.datasets import KarateClub
  
          return KarateClub(*args, **kwargs)
      if name.lower() in ["cora", "citeseer", "pubmed"]:
          from torch_geometric.datasets import Planetoid
  
          path = osp.join(root, "Planetoid", name)
          return Planetoid(path, name, *args, **kwargs)
      if name in ["BZR", "ENZYMES", "IMDB-BINARY", "MUTAG"]:
          from torch_geometric.datasets import TUDataset
  
          path = osp.join(root, "TUDataset")
          return TUDataset(path, name, *args, **kwargs)
      if name in ["ego-facebook", "soc-Slashdot0811", "wiki-vote"]:
          from torch_geometric.datasets import SNAPDataset
  
          path = osp.join(root, "SNAPDataset")
          return SNAPDataset(path, name, *args, **kwargs)
      if name.lower() in ["bashapes"]:
          from torch_geometric.datasets import BAShapes
  
          return BAShapes(*args, **kwargs)
      if name in ["citationCiteseer", "illc1850"]:
          from torch_geometric.datasets import (
              SuiteSparseMatrixCollection,
          )
  
          path = osp.join(root, "SuiteSparseMatrixCollection")
          return SuiteSparseMatrixCollection(
              path, name=name, *args, **kwargs
          )
      if "elliptic" in name.lower():
          from torch_geometric.datasets import EllipticBitcoinDataset
  
          path = osp.join(root, "EllipticBitcoinDataset")
          return EllipticBitcoinDataset(path, *args, **kwargs)
      if name.lower() in ["hetero"]:
          from torch_geometric.testing import FakeHeteroDataset
  
          return FakeHeteroDataset(*args, **kwargs)
  
      raise ValueError(f"Cannot load dataset with name '{name}'")
  
file_path: pytorch_geometric/test/conftest.py
imports: import functools
  import logging
  import os.path
  from typing import Callable
  import pytest
  import torch
  import torch_geometric.typing
  from torch_geometric.data import Dataset
  from torch_geometric.io import fs
label_desc: Use image node_id 1 for calling a global function with example usage: load_dataset(root, name) and returns: KarateClub, Planetoid, TUDataset, SNAPDataset, BAShapes, SuiteSparseMatrixCollection, EllipticBitcoinDataset, FakeHeteroDataset