code: def _get_sampler(sampler_type):
      if sampler_type == SamplerType.Normal:
          return gb.NeighborSampler
      if sampler_type == SamplerType.Layer:
          return gb.LayerNeighborSampler
      return partial(
          gb.TemporalNeighborSampler,
          node_timestamp_attr_name="timestamp",
          edge_timestamp_attr_name="timestamp",
      )
  
imports: import unittest
  from enum import Enum
  from functools import partial
  import backend
  import dgl
  import dgl.graphbolt
  import pytest
  import torch
  from torchdata.datapipes.iter import Mapper
  from .None import gb_test_utils
label_desc: Use image node_id 2 for calling a global function with example usage: _get_sampler(sampler_type) and returns: partial, gb, gb