code: def _sa_TIMESTAMP():
      class TIMESTAMP(sa.TIMESTAMP):
          def __init__(self, *args, timezone=True, **kwargs):
              super().__init__(*args, timezone=timezone, **kwargs)
  
      return TIMESTAMP
  
file_path: airflow/airflow/migrations/db_types.py
imports: from __future__ import annotations
  import sqlalchemy
  from alembic import context
  from lazy_object_proxy import Proxy
label_desc: Use image node_id 3 for calling a global function with example usage: _sa_TIMESTAMP() and returns: TIMESTAMP