code: def upgrade():
      bind = op.get_bind()
      session = db.Session(bind=bind)
  
      for datasource in session.query(SqlaTable):
          if datasource.extra:
              try:
                  extra = json.loads(datasource.extra)
  
                  if extra and "health_check" in extra:
                      del extra["health_check"]
                      datasource.extra = (
                          json.dumps(extra) if extra else None
                      )
              except Exception as ex:
                  logging.exception(ex)
  
      session.commit()
      session.close()
  
file_path: superset/superset/migrations/versions/2021-04-07_07-21_134cea61c5e7_remove_dataset_health_check_message.py
imports: import json
  import logging
  from alembic import op
  from sqlalchemy import Column, Integer, Text
  from sqlalchemy.ext.declarative import declarative_base
  from superset import db
label_desc: Use image node_id 1 for calling a global function with example usage: upgrade() without return types