code: def post_migration():
      with connect_to_mlflow_db() as conn:
          for table in TABLES:
              df_actual = pd.read_sql(
                  sa.text(f"SELECT * FROM {table}"), conn
              )
              df_expected = pd.read_pickle(
                  SNAPSHOTS_DIR / f"{table}.pkl"
              )
              pd.testing.assert_frame_equal(
                  df_actual[df_expected.columns], df_expected
              )
  
file_path: mlflow/tests/db/check_migration.py
imports: import os
  import uuid
  from pathlib import Path
  import click
  import pandas
  import sqlalchemy
  import mlflow
  from mlflow.store.model_registry.dbmodels.models import SqlModelVersion, SqlModelVersionTag, SqlRegisteredModel, SqlRegisteredModelTag
  from mlflow.store.tracking.dbmodels.models import SqlExperiment, SqlExperimentTag, SqlLatestMetric, SqlMetric, SqlParam, SqlRun, SqlTag
label_desc: Use image node_id 5 for calling a global function with example usage: post_migration() without return types