sqlmesh 0.188.2.dev17__py3-none-any.whl → 0.188.2.dev18__py3-none-any.whl

sqlmesh/core/test/definition.py

@@ -11,9 +11,7 @@ from pathlib import Path
 from unittest.mock import patch
 
 import numpy as np
-import pandas as pd
 from io import StringIO
-from pandas.api.types import is_object_dtype
 from sqlglot import Dialect, exp
 from sqlglot.optimizer.annotate_types import annotate_types
 from sqlglot.optimizer.normalize_identifiers import normalize_identifiers
@@ -29,6 +27,8 @@ from sqlmesh.utils.errors import ConfigError, TestError
 from sqlmesh.utils.yaml import load as yaml_load
 
 if t.TYPE_CHECKING:
+    import pandas as pd
+
     from sqlglot.dialects.dialect import DialectType
 
     Row = t.Dict[str, t.Any]
@@ -207,6 +207,9 @@ class ModelTest(unittest.TestCase):
         partial: t.Optional[bool] = False,
     ) -> None:
         """Compare two DataFrames"""
+        import pandas as pd
+        from pandas.api.types import is_object_dtype
+
         if partial:
             intersection = actual[actual.columns.intersection(expected.columns)]
             if len(intersection.columns) > 0:
@@ -385,6 +388,8 @@ class ModelTest(unittest.TestCase):
             partial: bool = False,
             dialect: DialectType = None,
         ) -> t.Dict:
+            import pandas as pd
+
             if not isinstance(values, dict):
                 values = {"rows": values}
 
@@ -557,6 +562,8 @@ class ModelTest(unittest.TestCase):
         columns: t.Optional[t.Collection] = None,
         partial: t.Optional[bool] = False,
     ) -> pd.DataFrame:
+        import pandas as pd
+
         query = values.get("query")
         if query:
             if not partial:
@@ -725,6 +732,8 @@ class PythonModelTest(ModelTest):
 
     def _execute_model(self) -> pd.DataFrame:
         """Executes the python model and returns a DataFrame."""
+        import pandas as pd
+
         with self._concurrent_render_context():
             variables = self.body.get("vars", {}).copy()
             time_kwargs = {key: variables.pop(key) for key in TIME_KWARG_KEYS if key in variables}
@@ -893,6 +902,8 @@ def _raise_if_unexpected_columns(
 
 def _row_difference(left: pd.DataFrame, right: pd.DataFrame) -> pd.DataFrame:
     """Returns all rows in `left` that don't appear in `right`."""
+    import pandas as pd
+
     rows_missing_from_right = []
 
     # `None` replaces `np.nan` because `np.nan != np.nan` and this would affect the mapping lookup

sqlmesh/dbt/adapter.py

@@ -4,7 +4,6 @@ import abc
 import logging
 import typing as t
 
-import pandas as pd
 from sqlglot import exp, parse_one
 
 from sqlmesh.core.dialect import normalize_and_quote, normalize_model_name, schema_
@@ -15,6 +14,7 @@ from sqlmesh.utils.jinja import JinjaMacroRegistry
 
 if t.TYPE_CHECKING:
     import agate
+    import pandas as pd
     from dbt.adapters.base import BaseRelation
     from dbt.adapters.base.column import Column
     from dbt.adapters.base.impl import AdapterResponse
@@ -324,6 +324,7 @@ class RuntimeAdapter(BaseAdapter):
     def execute(
         self, sql: str, auto_begin: bool = False, fetch: bool = False
     ) -> t.Tuple[AdapterResponse, agate.Table]:
+        import pandas as pd
         from dbt.adapters.base.impl import AdapterResponse
 
         from sqlmesh.dbt.util import pandas_to_agate, empty_table

sqlmesh/dbt/util.py

@@ -1,9 +1,13 @@
+from __future__ import annotations
+
 import typing as t
 
 import agate
-import pandas as pd
 from dbt.version import get_installed_version
 
+if t.TYPE_CHECKING:
+    import pandas as pd
+
 
 def _get_dbt_version() -> t.Tuple[int, int, int]:
     dbt_version = get_installed_version()

sqlmesh/migrations/v0007_env_table_info_to_kind.py

@@ -3,7 +3,6 @@
 import json
 import zlib
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -14,6 +13,8 @@ def _hash(data):  # type: ignore
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     environments_table = "_environments"

sqlmesh/migrations/v0009_remove_pre_post_hooks.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0011_add_model_kind_name.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0012_update_jinja_expressions.py

@@ -3,7 +3,6 @@
 import json
 import typing as t
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.jinja import has_jinja
@@ -11,6 +10,8 @@ from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0013_serde_using_model_dialects.py

@@ -3,7 +3,6 @@
 import json
 import typing as t
 
-import pandas as pd
 from sqlglot import exp, parse_one
 
 from sqlmesh.utils.jinja import has_jinja
@@ -11,6 +10,8 @@ from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0016_fix_windows_path.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0017_fix_windows_seed_path.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0018_rename_snapshot_model_to_node.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0020_remove_redundant_attributes_from_dbt_models.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0021_fix_table_properties.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.core import dialect as d
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0022_move_project_to_model.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0024_replace_model_kind_name_enum_with_value.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0025_fix_intervals_and_missing_change_category.py

@@ -3,7 +3,6 @@
 import json
 import zlib
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils import random_id
@@ -12,6 +11,8 @@ from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0026_remove_dialect_from_seed.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0027_minute_interval_to_five.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0029_generate_schema_types_using_dialect.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp, parse_one
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0030_update_unrestorable_snapshots.py

@@ -4,13 +4,14 @@ import json
 import typing as t
 from collections import defaultdict
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync: t.Any, **kwargs: t.Any) -> None:  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0031_remove_dbt_target_fields.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0034_add_default_catalog.py

@@ -5,7 +5,6 @@ from __future__ import annotations
 import json
 import typing as t
 
-import pandas as pd
 from sqlglot import exp
 from sqlglot.dialects.dialect import DialectType
 from sqlglot.helper import dict_depth, seq_get
@@ -65,6 +64,8 @@ def update_dbt_relations(
 
 
 def migrate(state_sync, default_catalog: t.Optional[str], **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0037_remove_dbt_is_incremental_macro.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0038_add_expiration_ts_to_snapshot.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.date import to_datetime, to_timestamp
@@ -11,6 +10,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0039_include_environment_in_plan_dag_spec.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     plan_dags_table = "_plan_dags"

sqlmesh/migrations/v0041_remove_hash_raw_query_attribute.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0042_trim_indirect_versions.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0043_fix_remove_obsolete_attributes_in_plan_dags.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     plan_dags_table = "_plan_dags"

sqlmesh/migrations/v0045_move_gateway_variable.py

@@ -3,7 +3,6 @@
 import ast
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -11,6 +10,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0048_drop_indirect_versions.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0051_rename_column_descriptions.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
@@ -10,6 +9,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0055_add_updated_ts_unpaused_ts_ttl_ms_unrestorable_to_snapshot.py

@@ -2,7 +2,6 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.date import to_datetime, to_timestamp
@@ -11,6 +10,8 @@ from sqlmesh.utils.migration import blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0060_move_audits_to_model.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0063_change_signals.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp, parse_one
 
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0064_join_when_matched_strings.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"

sqlmesh/migrations/v0069_update_dev_table_suffix.py

@@ -2,13 +2,14 @@
 
 import json
 
-import pandas as pd
 from sqlglot import exp
 
 from sqlmesh.utils.migration import index_text_type, blob_text_type
 
 
 def migrate(state_sync, **kwargs):  # type: ignore
+    import pandas as pd
+
     engine_adapter = state_sync.engine_adapter
     schema = state_sync.schema
     snapshots_table = "_snapshots"