sqlmesh 0.213.1.dev1__py3-none-any.whl → 0.227.2.dev4__py3-none-any.whl

sqlmesh/dbt/test.py

@@ -8,11 +8,11 @@ from pathlib import Path
 from pydantic import Field
 import sqlmesh.core.dialect as d
 from sqlmesh.core.audit import Audit, ModelAudit, StandaloneAudit
+from sqlmesh.core.node import DbtNodeInfo
 from sqlmesh.dbt.common import (
     Dependencies,
     GeneralConfig,
     SqlStr,
-    extract_jinja_config,
     sql_str_validator,
 )
 from sqlmesh.utils import AttributeDict
@@ -61,6 +61,10 @@ class TestConfig(GeneralConfig):
         error_if: Conditional expression (default "!=0") to detect if error condition met (Not supported).
     """
 
+    __test__ = (
+        False  # prevent pytest trying to collect this as a test class when it's imported in a test
+    )
+
     # SQLMesh fields
     path: Path = Path()
     name: str
@@ -76,8 +80,10 @@ class TestConfig(GeneralConfig):
     dialect_: t.Optional[str] = Field(None, alias="dialect")
 
     # dbt fields
+    unique_id: str = ""
     package_name: str = ""
     alias: t.Optional[str] = None
+    fqn: t.List[str] = []
     schema_: t.Optional[str] = Field("", alias="schema")
     database: t.Optional[str] = None
     severity: Severity = Severity.ERROR
@@ -103,9 +109,28 @@ class TestConfig(GeneralConfig):
     def _lowercase_name(cls, v: str) -> str:
         return v.lower()
 
+    @property
+    def canonical_name(self) -> str:
+        return f"{self.package_name}.{self.name}".lower() if self.package_name else self.name
+
     @property
     def is_standalone(self) -> bool:
-        return not self.model_name
+        # A test is standalone if:
+        # 1. It has no model_name (already standalone), OR
+        # 2. It references other models besides its own model
+        if not self.model_name:
+            return True
+
+        # Check if test has references to other models
+        # For versioned models, refs include version (e.g., "model_name_v1") but model_name may not
+        self_refs = {self.model_name}
+        for ref in self.dependencies.refs:
+            # versioned models end in _vX
+            if ref.startswith(f"{self.model_name}_v"):
+                self_refs.add(ref)
+
+        other_refs = {ref for ref in self.dependencies.refs if ref not in self_refs}
+        return bool(other_refs)
 
     @property
     def sqlmesh_config_fields(self) -> t.Set[str]:
@@ -134,9 +159,7 @@ class TestConfig(GeneralConfig):
             }
         )
 
-        sql_no_config, _sql_config_only = extract_jinja_config(self.sql)
-        sql_no_config = sql_no_config.replace("**_dbt_generic_test_kwargs", self._kwargs())
-        query = d.jinja_query(sql_no_config)
+        query = d.jinja_query(self.sql.replace("**_dbt_generic_test_kwargs", self._kwargs()))
 
         skip = not self.enabled
         blocking = self.severity == Severity.ERROR
@@ -146,6 +169,7 @@ class TestConfig(GeneralConfig):
             jinja_macros.add_globals({"this": self.relation_info})
             audit = StandaloneAudit(
                 name=self.name,
+                dbt_node_info=self.node_info,
                 dialect=self.dialect(context),
                 skip=skip,
                 query=query,
@@ -162,6 +186,7 @@ class TestConfig(GeneralConfig):
         else:
             audit = ModelAudit(
                 name=self.name,
+                dbt_node_info=self.node_info,
                 dialect=self.dialect(context),
                 skip=skip,
                 blocking=blocking,
@@ -205,6 +230,12 @@ class TestConfig(GeneralConfig):
             }
         )
 
+    @property
+    def node_info(self) -> DbtNodeInfo:
+        return DbtNodeInfo(
+            unique_id=self.unique_id, name=self.name, fqn=".".join(self.fqn), alias=self.alias
+        )
+
 
 def _remove_jinja_braces(jinja_str: str) -> str:
     no_braces = jinja_str

sqlmesh/migrations/v0000_baseline.py

@@ -0,0 +1,95 @@
+"""The baseline migration script that sets up the initial state tables."""
+
+from sqlglot import exp
+from sqlmesh.utils.migration import blob_text_type, index_text_type
+
+
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    intervals_table = "_intervals"
+    snapshots_table = "_snapshots"
+    environments_table = "_environments"
+    versions_table = "_versions"
+    if schema:
+        engine_adapter.create_schema(schema)
+        intervals_table = f"{schema}.{intervals_table}"
+        snapshots_table = f"{schema}.{snapshots_table}"
+        environments_table = f"{schema}.{environments_table}"
+        versions_table = f"{schema}.{versions_table}"
+
+    index_type = index_text_type(engine_adapter.dialect)
+    blob_type = blob_text_type(engine_adapter.dialect)
+
+    snapshots_columns_to_types = {
+        "name": exp.DataType.build(index_type),
+        "identifier": exp.DataType.build(index_type),
+        "version": exp.DataType.build(index_type),
+        "snapshot": exp.DataType.build(blob_type),
+        "kind_name": exp.DataType.build(index_type),
+        "updated_ts": exp.DataType.build("bigint"),
+        "unpaused_ts": exp.DataType.build("bigint"),
+        "ttl_ms": exp.DataType.build("bigint"),
+        "unrestorable": exp.DataType.build("boolean"),
+    }
+
+    environments_columns_to_types = {
+        "name": exp.DataType.build(index_type),
+        "snapshots": exp.DataType.build(blob_type),
+        "start_at": exp.DataType.build("text"),
+        "end_at": exp.DataType.build("text"),
+        "plan_id": exp.DataType.build("text"),
+        "previous_plan_id": exp.DataType.build("text"),
+        "expiration_ts": exp.DataType.build("bigint"),
+        "finalized_ts": exp.DataType.build("bigint"),
+        "promoted_snapshot_ids": exp.DataType.build(blob_type),
+        "suffix_target": exp.DataType.build("text"),
+        "catalog_name_override": exp.DataType.build("text"),
+        "previous_finalized_snapshots": exp.DataType.build(blob_type),
+        "normalize_name": exp.DataType.build("boolean"),
+        "requirements": exp.DataType.build(blob_type),
+    }
+
+    intervals_columns_to_types = {
+        "id": exp.DataType.build(index_type),
+        "created_ts": exp.DataType.build("bigint"),
+        "name": exp.DataType.build(index_type),
+        "identifier": exp.DataType.build(index_type),
+        "version": exp.DataType.build(index_type),
+        "start_ts": exp.DataType.build("bigint"),
+        "end_ts": exp.DataType.build("bigint"),
+        "is_dev": exp.DataType.build("boolean"),
+        "is_removed": exp.DataType.build("boolean"),
+        "is_compacted": exp.DataType.build("boolean"),
+    }
+
+    versions_columns_to_types = {
+        "schema_version": exp.DataType.build("int"),
+        "sqlglot_version": exp.DataType.build(index_type),
+        "sqlmesh_version": exp.DataType.build(index_type),
+    }
+
+    # Create the versions table.
+    engine_adapter.create_state_table(versions_table, versions_columns_to_types)
+
+    # Create the snapshots table and its indexes.
+    engine_adapter.create_state_table(
+        snapshots_table, snapshots_columns_to_types, primary_key=("name", "identifier")
+    )
+    engine_adapter.create_index(snapshots_table, "_snapshots_name_version_idx", ("name", "version"))
+
+    # Create the environments table and its indexes.
+    engine_adapter.create_state_table(
+        environments_table, environments_columns_to_types, primary_key=("name",)
+    )
+
+    # Create the intervals table and its indexes.
+    engine_adapter.create_state_table(
+        intervals_table, intervals_columns_to_types, primary_key=("id",)
+    )
+    engine_adapter.create_index(
+        intervals_table, "_intervals_name_identifier_idx", ("name", "identifier")
+    )
+    engine_adapter.create_index(intervals_table, "_intervals_name_version_idx", ("name", "version"))
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
+    pass

sqlmesh/migrations/v0061_mysql_fix_blob_text_type.py

@@ -9,20 +9,15 @@ from sqlglot import exp
 from sqlmesh.utils.migration import blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
     if engine_adapter.dialect != "mysql":
         return
-
-    schema = state_sync.schema
     environments_table = "_environments"
     snapshots_table = "_snapshots"
-    plan_dags_table = "_plan_dags"
 
     if schema:
         environments_table = f"{schema}.{environments_table}"
         snapshots_table = f"{schema}.{snapshots_table}"
-        plan_dags_table = f"{schema}.{plan_dags_table}"
 
     targets = [
         (environments_table, "snapshots"),
@@ -30,7 +25,6 @@ def migrate(state_sync, **kwargs):  # type: ignore
         (environments_table, "previous_finalized_snapshots"),
         (environments_table, "requirements"),
         (snapshots_table, "snapshot"),
-        (plan_dags_table, "dag_spec"),
     ]
 
     for table_name, column_name in targets:
@@ -47,3 +41,7 @@ def migrate(state_sync, **kwargs):  # type: ignore
         )
 
         engine_adapter.execute(alter_table_exp)
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
+    pass

sqlmesh/migrations/v0062_add_model_gateway.py

@@ -1,5 +1,9 @@
 """Add the gateway model attribute."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0063_change_signals.py

@@ -7,11 +7,13 @@ from sqlglot import exp, parse_one
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     index_type = index_text_type(engine_adapter.dialect)
     if schema:

sqlmesh/migrations/v0064_join_when_matched_strings.py

@@ -7,11 +7,13 @@ from sqlglot import exp
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     index_type = index_text_type(engine_adapter.dialect)
     if schema:

sqlmesh/migrations/v0065_add_model_optimize.py

@@ -1,5 +1,9 @@
 """Add the optimize_query model attribute."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0066_add_auto_restatements.py

@@ -5,9 +5,7 @@ from sqlglot import exp
 from sqlmesh.utils.migration import index_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
     auto_restatements_table = "_auto_restatements"
     intervals_table = "_intervals"
 
@@ -39,6 +37,13 @@ def migrate(state_sync, **kwargs):  # type: ignore
     )
     engine_adapter.execute(alter_table_exp)
 
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
+    intervals_table = "_intervals"
+
+    if schema:
+        intervals_table = f"{schema}.{intervals_table}"
+
     engine_adapter.update_table(
         intervals_table,
         {"is_pending_restatement": False},

sqlmesh/migrations/v0067_add_tsql_date_full_precision.py

@@ -1,5 +1,9 @@
 """Add full precision for tsql to support nanoseconds."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0068_include_unrendered_query_in_metadata_hash.py

@@ -1,5 +1,9 @@
 """Include the unrendered query in the metadata hash."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0069_update_dev_table_suffix.py

@@ -7,11 +7,13 @@ from sqlglot import exp
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     environments_table = "_environments"
     if schema:

sqlmesh/migrations/v0070_include_grains_in_metadata_hash.py

@@ -1,5 +1,9 @@
 """Include grains in the metadata hash."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0071_add_dev_version_to_intervals.py

@@ -8,14 +8,10 @@ from sqlglot import exp
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
     intervals_table = "_intervals"
-    snapshots_table = "_snapshots"
     if schema:
         intervals_table = f"{schema}.{intervals_table}"
-        snapshots_table = f"{schema}.{snapshots_table}"
 
     index_type = index_text_type(engine_adapter.dialect)
     alter_table_exp = exp.Alter(
@@ -30,6 +26,14 @@ def migrate(state_sync, **kwargs):  # type: ignore
     )
     engine_adapter.execute(alter_table_exp)
 
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
+    intervals_table = "_intervals"
+    snapshots_table = "_snapshots"
+    if schema:
+        intervals_table = f"{schema}.{intervals_table}"
+        snapshots_table = f"{schema}.{snapshots_table}"
+
     used_dev_versions: t.Set[t.Tuple[str, str]] = set()
     used_versions: t.Set[t.Tuple[str, str]] = set()
     used_snapshot_ids: t.Set[t.Tuple[str, str]] = set()

sqlmesh/migrations/v0072_add_environment_statements.py

@@ -5,9 +5,7 @@ from sqlglot import exp
 from sqlmesh.utils.migration import blob_text_type, index_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
     environment_statements_table = "_environment_statements"
 
     if schema:
@@ -25,3 +23,7 @@ def migrate(state_sync, **kwargs):  # type: ignore
         },
         primary_key=("environment_name",),
     )
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
+    pass

sqlmesh/migrations/v0073_remove_symbolic_disable_restatement.py

@@ -6,11 +6,13 @@ from sqlglot import exp
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     if schema:
         snapshots_table = f"{schema}.{snapshots_table}"

sqlmesh/migrations/v0074_add_partition_by_time_column_property.py

@@ -2,5 +2,9 @@
 (default: True to keep the original behaviour)"""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0075_remove_validate_query.py

@@ -8,11 +8,13 @@ from sqlmesh.utils.migration import index_text_type
 from sqlmesh.utils.migration import blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     index_type = index_text_type(engine_adapter.dialect)
     if schema:

sqlmesh/migrations/v0076_add_cron_tz.py

@@ -1,5 +1,9 @@
 """Add 'cron_tz' property to node definition."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0077_fix_column_type_hash_calculation.py

@@ -1,5 +1,9 @@
 """Use the model's dialect when calculating the hash for the column types."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0078_warn_if_non_migratable_python_env.py

@@ -24,9 +24,11 @@ import sqlmesh.core.dialect as d
 from sqlmesh.core.console import get_console
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     snapshots_table = "_snapshots"
     if schema:
         snapshots_table = f"{schema}.{snapshots_table}"

sqlmesh/migrations/v0079_add_gateway_managed_property.py

@@ -3,11 +3,10 @@
 from sqlglot import exp
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
     environments_table = "_environments"
-    if state_sync.schema:
-        environments_table = f"{state_sync.schema}.{environments_table}"
+    if schema:
+        environments_table = f"{schema}.{environments_table}"
 
     alter_table_exp = exp.Alter(
         this=exp.to_table(environments_table),
@@ -21,7 +20,13 @@ def migrate(state_sync, **kwargs):  # type: ignore
     )
     engine_adapter.execute(alter_table_exp)
 
-    state_sync.engine_adapter.update_table(
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
+    environments_table = "_environments"
+    if schema:
+        environments_table = f"{schema}.{environments_table}"
+
+    engine_adapter.update_table(
         environments_table,
         {"gateway_managed": False},
         where=exp.true(),

sqlmesh/migrations/v0080_add_batch_size_to_scd_type_2_models.py

@@ -1,5 +1,9 @@
 """Add batch_size to SCD Type 2 models and add updated_at_name to by time which changes their data hash."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0081_update_partitioned_by.py

@@ -8,11 +8,13 @@ from sqlmesh.utils.migration import index_text_type
 from sqlmesh.utils.migration import blob_text_type
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     index_type = index_text_type(engine_adapter.dialect)
     if schema:

sqlmesh/migrations/v0082_warn_if_incorrectly_duplicated_statements.py

@@ -34,9 +34,11 @@ from sqlglot import exp
 from sqlmesh.core.console import get_console
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     snapshots_table = "_snapshots"
     if schema:
         snapshots_table = f"{schema}.{snapshots_table}"

sqlmesh/migrations/v0083_use_sql_for_scd_time_data_type_data_hash.py

@@ -1,5 +1,9 @@
 """Use sql(...) instead of gen when computing the data hash of the time data type."""
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0084_normalize_quote_when_matched_and_merge_filter.py

@@ -5,5 +5,9 @@ prevent un-normalized identifiers being quoted at the EngineAdapter level
 """
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     pass

sqlmesh/migrations/v0085_deterministic_repr.py

@@ -36,11 +36,13 @@ def _dict_sort(obj: t.Any) -> str:
     return repr(obj)
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     if schema:
         snapshots_table = f"{schema}.{snapshots_table}"

sqlmesh/migrations/v0086_check_deterministic_bug.py

@@ -10,9 +10,11 @@ logger = logging.getLogger(__name__)
 KEYS_TO_MAKE_DETERMINISTIC = ["__sqlmesh__vars__", "__sqlmesh__blueprint__vars__"]
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     snapshots_table = "_snapshots"
     versions_table = "_versions"
     if schema:

sqlmesh/migrations/v0087_normalize_blueprint_variables.py

@@ -35,11 +35,13 @@ class SqlValue:
     sql: str
 
 
-def migrate(state_sync, **kwargs):  # type: ignore
+def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
+    pass
+
+
+def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
     import pandas as pd
 
-    engine_adapter = state_sync.engine_adapter
-    schema = state_sync.schema
     snapshots_table = "_snapshots"
     if schema:
         snapshots_table = f"{schema}.{snapshots_table}"