code: def map_date_type(data_type):
      """Map column date type to pyarrow date type."""
      kind, bit_width, f_string, _ = data_type
  
      if kind == DtypeKind.DATETIME:
          unit, tz = parse_datetime_format_str(f_string)
          return pa.timestamp(unit, tz=tz)
      else:
          pa_dtype = _PYARROW_DTYPES.get(kind, {}).get(bit_width, None)
  
          # Error if dtype is not supported
          if pa_dtype:
              return pa_dtype
          else:
              raise NotImplementedError(
                  f"Conversion for {data_type} is not yet supported."
              )
  
file_path: vegafusion/python/vegafusion/vegafusion/datasource/dfi_datasource.py
imports: from typing import Iterable
  import re
  import pyarrow
  from typing import Any, Dict
  from ._dfi_types import DtypeKind, DataFrame
  from .datasource import Datasource
label_desc: Use image node_id 2 for calling a global function with example usage: map_date_type(data_type) and returns: pa, pa_dtype