clickzetta-semantic-model-generator 1.0.2__py3-none-any.whl → 1.0.3__py3-none-any.whl

{clickzetta_semantic_model_generator-1.0.2.dist-info → clickzetta_semantic_model_generator-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: clickzetta-semantic-model-generator
-Version: 1.0.2
+Version: 1.0.3
 Summary: Curate a Semantic Model for ClickZetta Lakehouse
 License: Apache Software License; BSD License
 Author: qililiang
@@ -13,12 +13,12 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Provides-Extra: looker
 Requires-Dist: PyYAML (>=6.0.1,<7.0.0)
-Requires-Dist: clickzetta-connector-python (==0.8.92)
-Requires-Dist: clickzetta-zettapark-python (==0.1.3)
+Requires-Dist: clickzetta-connector-python (>=0.8.92)
+Requires-Dist: clickzetta-zettapark-python (>=0.1.3)
 Requires-Dist: dashscope (>=1.22.2,<2.0.0)
 Requires-Dist: loguru (>=0.7.2,<0.8.0)
 Requires-Dist: looker-sdk (>=24.14.0,<25.0.0) ; extra == "looker"
-Requires-Dist: numpy (>=1.26.4,<2.0.0)
+Requires-Dist: numpy (>=1.26.4,<3.0.0)
 Requires-Dist: pandas (>=2.0.1,<3.0.0)
 Requires-Dist: protobuf (==5.26.1)
 Requires-Dist: pyarrow (==14.0.2)
@@ -31,7 +31,7 @@ Requires-Dist: streamlit (==1.36.0)
 Requires-Dist: streamlit-extras (==0.4.0)
 Requires-Dist: strictyaml (>=1.7.3,<2.0.0)
 Requires-Dist: tqdm (>=4.66.5,<5.0.0)
-Requires-Dist: urllib3 (>=1.26.19,<2.0.0)
+Requires-Dist: urllib3 (>=1.26.19,<3.0.0)
 Description-Content-Type: text/markdown
 
 # semantic-model-generator

{clickzetta_semantic_model_generator-1.0.2.dist-info → clickzetta_semantic_model_generator-1.0.3.dist-info}/RECORD

@@ -1,38 +1,38 @@
 semantic_model_generator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-semantic_model_generator/clickzetta_utils/clickzetta_connector.py,sha256=hnXBtwlZbS76e1H6w8dkVN1-WCd9CeVinoPVJiD4aJk,32335
-semantic_model_generator/clickzetta_utils/env_vars.py,sha256=01AQV7WXByZBBAHETEEQ22mRytKvzcge_yn9RZH09DE,7597
+semantic_model_generator/clickzetta_utils/clickzetta_connector.py,sha256=rFBWdNQerLYinn6RoDV_J4k2G4LLofiFkLDa7j8hmng,32888
+semantic_model_generator/clickzetta_utils/env_vars.py,sha256=8cbL6R75c1-aVQ2i1TDr9SiHCUjTrgvXbIRz4MbcmbE,7664
 semantic_model_generator/clickzetta_utils/utils.py,sha256=D0SX2faBjwvhFJLt1Yk4mlZmyHmQt7LN93Jrc5YIU-A,3800
 semantic_model_generator/data_processing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-semantic_model_generator/data_processing/cte_utils.py,sha256=FAsxbeAXYz1q0Iei_pDUKBrOWLnnH1jMd4h_5dXmoHE,16086
+semantic_model_generator/data_processing/cte_utils.py,sha256=jfTJIwc89-0nnelVw_5vpIVRout7V0YooUDfZzTzDr4,16086
 semantic_model_generator/data_processing/cte_utils_test.py,sha256=l6QkyyH22FexLKjvvbS9Je3YtdTrJE3a-BiknCy1g9s,2822
 semantic_model_generator/data_processing/data_types.py,sha256=1HsSCkdCWvcXiwN3o1-HVQi_ZVIR0lYevXG9CE1TvRc,1172
 semantic_model_generator/data_processing/proto_utils.py,sha256=UwqCfQYilTx68KcA4IYZN7PeM4Pz_pK1h0FrVJomzV8,2938
-semantic_model_generator/generate_model.py,sha256=awBj72QkrfCq9Ij-lo6_tNDt7yQ0mFEElVuz7yfPy1E,116381
+semantic_model_generator/generate_model.py,sha256=ogNvx1HNOnC5KIZlGDwcWL7PLMHRs8zcZZbwricffDo,121843
 semantic_model_generator/llm/__init__.py,sha256=rLQt2pzRmxtnBLKjxN_qZ2a_nvkFHtmguU5lyajCldw,1030
-semantic_model_generator/llm/dashscope_client.py,sha256=DBbxuqy1BIk9mxm20GSK-9rfDtohy3KRTsxpsG5sjf4,6031
-semantic_model_generator/llm/enrichment.py,sha256=8PlUFC2LtuSaQGdrJIoQpn_oGUyNV165B4FA3046CYI,40316
-semantic_model_generator/llm/progress_tracker.py,sha256=tbdbfCNv2wlh-N5wlcMa10JVcmbdkQojPVe-VPWmmxM,6757
+semantic_model_generator/llm/dashscope_client.py,sha256=lHS36iqNZbFhwgidPpW1Bwwy4S2O7GeLyMSMdlSoBsY,6050
+semantic_model_generator/llm/enrichment.py,sha256=49e9Jg_jHfhUIEQ3JserEc5DV5sFWA12K76TY4UwnCg,41448
+semantic_model_generator/llm/progress_tracker.py,sha256=XmQUDSmc8Uz3bdnV-K-SHmmSoJjxLp3SPMxXwVDKf88,6740
 semantic_model_generator/output_models/.keep,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 semantic_model_generator/protos/semantic_model.proto,sha256=WZiN4b8vR-ZX-Lj9Vsm6HjZNAyNvM1znIyut_YkPVSI,16473
 semantic_model_generator/protos/semantic_model_pb2.py,sha256=scbWkW-I-r3_hp_5SHoOWn02p52RJ9DJ0_-nRgr0LHc,25606
 semantic_model_generator/protos/semantic_model_pb2.pyi,sha256=iiBIZxtX9d6IuUO3aLcsJsHUeZqdi14vYNuUsSM8C0g,18267
 semantic_model_generator/relationships/__init__.py,sha256=HN6Opie25Oawt2fCDM_bZwRBVBEzqRsEXgDzYC7ytns,373
-semantic_model_generator/relationships/discovery.py,sha256=fy8mzfAZtWWVUjiSP5jeKffoa2GodNIKA3eGi4dxhHo,6020
-semantic_model_generator/tests/clickzetta_connector_test.py,sha256=NKDbhll8TxFtbwFJIUcQQDKJU1LkfEz9nL0SE32hx3o,3114
-semantic_model_generator/tests/cte_utils_test.py,sha256=8v2nrrD2GkH_PTGIsKm3lQ06unzis8iR31atT8bUX98,17385
-semantic_model_generator/tests/generate_model_classification_test.py,sha256=q7dh29h9iF17ChzuoFSLtRX9ASiAm2oY4OkGyFVfn5Y,2117
-semantic_model_generator/tests/llm_enrichment_test.py,sha256=YeYg4voQ3wy2vgF7H9JNdnMOyZmUfiMdL6oEXFv-ztg,14415
-semantic_model_generator/tests/relationship_discovery_test.py,sha256=1SVX59-mpHQvxk7RDGtglesg6VXU9TnnZZcfjZi5IHs,3448
-semantic_model_generator/tests/relationships_filters_test.py,sha256=fVyA-hwxGqdlFr_PGT9YdrPz13XGvI4J_5F3IpVTFEE,8009
+semantic_model_generator/relationships/discovery.py,sha256=l_CixbfRvHBqxmLCmCq7bvQHRt3iUl0o5mui4R5LHXQ,5961
+semantic_model_generator/tests/clickzetta_connector_test.py,sha256=Fdx7jooNt1lslKB2Ub51wqOZ8OM0osgZiDDl3bV6riw,3086
+semantic_model_generator/tests/cte_utils_test.py,sha256=LdhWw_bHZDE1LyS2hBVy_VTNjLgodonesWaxw8jXpV4,17385
+semantic_model_generator/tests/generate_model_classification_test.py,sha256=Amq29cmeKd0S7iVikJ60RFm9gpWaQv1TijXofp3J-lI,2275
+semantic_model_generator/tests/llm_enrichment_test.py,sha256=1avLrPWp7J7o_K3PKbI_PIvduM5Id21MmoL0JTeDTfs,15738
+semantic_model_generator/tests/relationship_discovery_test.py,sha256=SOuXCwbmSUgvZoOS2s5oGK1w0LW283M1hg--QlLaDVA,3490
+semantic_model_generator/tests/relationships_filters_test.py,sha256=bUm3r1UGaXca-hJOot7jMPz4It_TVsoddd-Xpk-76zM,10166
 semantic_model_generator/tests/samples/validate_yamls.py,sha256=262j-2i2oFZtTyK2susOrbxxE5eS-6IN-V0jFEOpt_w,156249
-semantic_model_generator/tests/utils_test.py,sha256=Tfvb-ErZPBS_HjXr4N7XSJnL3hlncNGF5pay1xFfNHg,539
+semantic_model_generator/tests/utils_test.py,sha256=HWRXR45QYL1f6L8xsMppqLXzF9HAsrMwTMQIKpZrc_M,539
 semantic_model_generator/tests/validate_model_test.py,sha256=kfG0dkhuDlJ47_Zt3seXPOV4DmzWOTTGXQINKvLYtlo,352
 semantic_model_generator/tests/yaml_to_semantic_model_test.py,sha256=sfjHahOLdn0xWnNn12e8K8TvyKl17WP4sLgSA1U9mB0,2626
 semantic_model_generator/validate/context_length.py,sha256=HL-GfaRXNcVji1-pAFGXGxMoH2fyQtQok7Y9ybXn6Tg,4454
-semantic_model_generator/validate/keywords.py,sha256=mFtmIK72YLQ7wODL-zCC-uMXzXMJT6Tg4fISAin2WtQ,5480
-semantic_model_generator/validate/schema.py,sha256=ImFUzO5EzNbl65-0dqt5AmdEvy4lqX8gBkOSOKN8uZM,5863
+semantic_model_generator/validate/keywords.py,sha256=frZ5HjRXP69K6dYAU5_d86oSp40_3yoLUg1eQwU3oLM,7080
+semantic_model_generator/validate/schema.py,sha256=eL_wl5yscIeczwNBRUKhF_7QqWW2wSGimkgaOhMFsrA,5893
 semantic_model_generator/validate_model.py,sha256=Uq-V-GfPeF2Dy4l9uF5Guv104gDCDGh0Cxz1AJOu5dk,836
-clickzetta_semantic_model_generator-1.0.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-clickzetta_semantic_model_generator-1.0.2.dist-info/METADATA,sha256=vgObGkoBnyfxUuJM8eAH8ZnIGuJrueJFnU1UfTo_gN4,7816
-clickzetta_semantic_model_generator-1.0.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-clickzetta_semantic_model_generator-1.0.2.dist-info/RECORD,,
+clickzetta_semantic_model_generator-1.0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+clickzetta_semantic_model_generator-1.0.3.dist-info/METADATA,sha256=A1kBc4PO_LEbIjWM-24jHnnV6NynmowuX5Jy91tlWBk,7816
+clickzetta_semantic_model_generator-1.0.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+clickzetta_semantic_model_generator-1.0.3.dist-info/RECORD,,

semantic_model_generator/clickzetta_utils/clickzetta_connector.py

@@ -4,7 +4,6 @@ import concurrent.futures
 import re
 from collections import defaultdict
 from contextlib import contextmanager
-from types import SimpleNamespace
 from typing import Any, Dict, Generator, List, Optional, TypeVar, Union
 
 import pandas as pd
@@ -12,10 +11,7 @@ from clickzetta.zettapark.session import Session
 from loguru import logger
 
 from semantic_model_generator.clickzetta_utils import env_vars
-from semantic_model_generator.clickzetta_utils.utils import (
-    clickzetta_connection,
-    create_session,
-)
+from semantic_model_generator.clickzetta_utils.utils import create_session
 from semantic_model_generator.data_processing.data_types import Column, Table
 
 ConnectionType = TypeVar("ConnectionType", bound=Session)
@@ -115,7 +111,9 @@ class ClickzettaCursor:
     def execute(self, query: str) -> "ClickzettaCursor":
         self._df = _execute_query_to_pandas(self._session, query)
         columns = [] if self._df is None else list(self._df.columns)
-        self.description = [(col, None, None, None, None, None, None) for col in columns]
+        self.description = [
+            (col, None, None, None, None, None, None) for col in columns
+        ]
         return self
 
     def fetchone(self) -> Optional[tuple[Any, ...]]:
@@ -159,7 +157,11 @@ def _quote_identifier(name: str) -> str:
 
 def _qualify_table(workspace: str, schema_name: str, table_name: str) -> str:
     return ".".join(
-        [_quote_identifier(workspace), _quote_identifier(schema_name), _quote_identifier(table_name)]
+        [
+            _quote_identifier(workspace),
+            _quote_identifier(schema_name),
+            _quote_identifier(table_name),
+        ]
     )
 
 
@@ -214,12 +216,18 @@ def _fetch_distinct_values(
     column_name: str,
     ndv: int,
 ) -> Optional[List[str]]:
-    workspace_part = _sanitize_identifier(workspace, workspace).upper() if workspace else ""
-    schema_part = _sanitize_identifier(schema_name, schema_name).upper() if schema_name else ""
+    workspace_part = (
+        _sanitize_identifier(workspace, workspace).upper() if workspace else ""
+    )
+    schema_part = (
+        _sanitize_identifier(schema_name, schema_name).upper() if schema_name else ""
+    )
     table_part = _sanitize_identifier(table_name, table_name).upper()
     column_part = _sanitize_identifier(column_name, column_name).upper()
 
-    qualified_parts = [part for part in (workspace_part, schema_part, table_part) if part]
+    qualified_parts = [
+        part for part in (workspace_part, schema_part, table_part) if part
+    ]
     qualified_table = ".".join(qualified_parts)
 
     query = f"SELECT DISTINCT {column_part} FROM {qualified_table} LIMIT {ndv}"
@@ -257,7 +265,6 @@ def _get_column_representation(
     else:
         column_datatype = str(column_datatype_raw)
     column_datatype = _normalize_column_type(column_datatype)
-    normalized_type = column_datatype.split("(")[0].strip()
     column_values = (
         _fetch_distinct_values(
             session=session,
@@ -351,7 +358,14 @@ def _catalog_category(session: Session, workspace: str) -> str:
         return "UNKNOWN"
 
     df.columns = [str(col).upper() for col in df.columns]
-    name_col = next((col for col in ("WORKSPACE_NAME", "NAME", "CATALOG_NAME") if col in df.columns), None)
+    name_col = next(
+        (
+            col
+            for col in ("WORKSPACE_NAME", "NAME", "CATALOG_NAME")
+            if col in df.columns
+        ),
+        None,
+    )
     category_col = next((col for col in ("CATEGORY",) if col in df.columns), None)
     if not name_col or not category_col:
         _CATALOG_CATEGORY_CACHE[workspace_upper] = "UNKNOWN"
@@ -408,7 +422,9 @@ ORDER BY kc.ordinal_position
         if result is not None:
             return result
     except Exception:
-        logger.debug("Primary key lookup via sys.information_schema failed; falling back.")
+        logger.debug(
+            "Primary key lookup via sys.information_schema failed; falling back."
+        )
 
     fallback_query = f"""
 SELECT kc.column_name
@@ -423,7 +439,13 @@ ORDER BY kc.ordinal_position
         if result is not None:
             return result
     except Exception as exc:
-        logger.warning("Primary key lookup failed for {}.{}.{}: {}", workspace, schema_name, table_name, exc)
+        logger.warning(
+            "Primary key lookup failed for {}.{}.{}: {}",
+            workspace,
+            schema_name,
+            table_name,
+            exc,
+        )
     return None
 
 
@@ -432,9 +454,7 @@ def _build_information_schema_query(
     table_schema: Optional[str],
     table_names: Optional[List[str]],
 ) -> str:
-    where_conditions: List[str] = [
-        "1=1"
-    ]
+    where_conditions: List[str] = ["1=1"]
     if table_schema:
         where_conditions.append(f"upper(t.table_schema) = '{table_schema.upper()}'")
     if table_names:
@@ -442,7 +462,6 @@ def _build_information_schema_query(
         where_conditions.append(f"upper(t.table_name) IN ({formatted_names})")
 
     where_clause = " AND ".join(where_conditions)
-    base = "information_schema"
     return f"""
 SELECT
     t.table_schema AS {_TABLE_SCHEMA_COL},
@@ -474,27 +493,48 @@ def _fetch_columns_via_show(
     schema = table_schema.upper() if table_schema else ""
 
     for table_name in table_names:
-        qualified_parts = [part for part in (catalog, schema, table_name.upper()) if part]
+        qualified_parts = [
+            part for part in (catalog, schema, table_name.upper()) if part
+        ]
         qualified_table = ".".join(qualified_parts)
         query = f"SHOW COLUMNS IN {qualified_table}"
         try:
             df = session.sql(query).to_pandas()
         except Exception as exc:
-            logger.debug("SHOW COLUMNS fallback failed for {}: {}", qualified_table, exc)
+            logger.debug(
+                "SHOW COLUMNS fallback failed for {}: {}", qualified_table, exc
+            )
             continue
         if df.empty:
             continue
         df.columns = [str(col).upper() for col in df.columns]
-        schema_col = next((col for col in ("TABLE_SCHEMA", "SCHEMA_NAME") if col in df.columns), None)
-        table_col = next((col for col in ("TABLE_NAME", "NAME") if col in df.columns), None)
-        column_col = next((col for col in ("COLUMN_NAME", "NAME") if col in df.columns and col != table_col), None)
-        datatype_col = next((col for col in ("DATA_TYPE", "TYPE") if col in df.columns), None)
-        comment_col = next((col for col in ("COMMENT", "COLUMN_COMMENT") if col in df.columns), None)
+        schema_col = next(
+            (col for col in ("TABLE_SCHEMA", "SCHEMA_NAME") if col in df.columns), None
+        )
+        table_col = next(
+            (col for col in ("TABLE_NAME", "NAME") if col in df.columns), None
+        )
+        column_col = next(
+            (
+                col
+                for col in ("COLUMN_NAME", "NAME")
+                if col in df.columns and col != table_col
+            ),
+            None,
+        )
+        datatype_col = next(
+            (col for col in ("DATA_TYPE", "TYPE") if col in df.columns), None
+        )
+        comment_col = next(
+            (col for col in ("COMMENT", "COLUMN_COMMENT") if col in df.columns), None
+        )
 
         normalized = pd.DataFrame()
         normalized[_TABLE_SCHEMA_COL] = df[schema_col] if schema_col else table_schema
         normalized[_TABLE_NAME_COL] = df[table_col] if table_col else table_name
-        normalized[_COLUMN_NAME_COL] = df[column_col] if column_col else df.index.astype(str)
+        normalized[_COLUMN_NAME_COL] = (
+            df[column_col] if column_col else df.index.astype(str)
+        )
         normalized[_DATATYPE_COL] = df[datatype_col] if datatype_col else ""
         normalized[_COLUMN_COMMENT_ALIAS] = df[comment_col] if comment_col else ""
         normalized[_TABLE_COMMENT_COL] = ""
@@ -552,6 +592,7 @@ def get_valid_schemas_tables_columns_df(
     if _TABLE_SCHEMA_COL in result.columns:
         result[_TABLE_SCHEMA_COL] = result[_TABLE_SCHEMA_COL].astype(str).str.upper()
     if _IS_PRIMARY_KEY_COL in result.columns:
+
         def _normalize_pk(value: Any) -> bool:
             if isinstance(value, bool):
                 return value
@@ -617,10 +658,11 @@ def fetch_tables_views_in_schema(
     workspace_upper = workspace.upper()
     schema_upper = schema.upper()
 
-    target = ""
     try:
         if workspace_upper and schema_upper:
-            df = session.sql(f"SHOW TABLES IN {workspace_upper}.{schema_upper}").to_pandas()
+            df = session.sql(
+                f"SHOW TABLES IN {workspace_upper}.{schema_upper}"
+            ).to_pandas()
         else:
             df = session.sql("SHOW TABLES").to_pandas()
     except Exception as exc:  # pragma: no cover
@@ -634,17 +676,27 @@ def fetch_tables_views_in_schema(
     df.columns = [str(col).upper() for col in df.columns]
     name_column = "TABLE_NAME" if "TABLE_NAME" in df.columns else df.columns[0]
     schema_column = next(
-        (col for col in ("SCHEMA_NAME", "TABLE_SCHEMA", "NAMESPACE") if col in df.columns),
+        (
+            col
+            for col in ("SCHEMA_NAME", "TABLE_SCHEMA", "NAMESPACE")
+            if col in df.columns
+        ),
         None,
     )
     catalog_column = next(
-        (col for col in ("CATALOG_NAME", "WORKSPACE_NAME", "TABLE_CATALOG") if col in df.columns),
+        (
+            col
+            for col in ("CATALOG_NAME", "WORKSPACE_NAME", "TABLE_CATALOG")
+            if col in df.columns
+        ),
         None,
     )
 
     results: List[str] = []
     for _, row in df.iterrows():
-        if _value_is_true(row.get("IS_VIEW")) and not _value_is_true(row.get("IS_MATERIALIZED_VIEW")):
+        if _value_is_true(row.get("IS_VIEW")) and not _value_is_true(
+            row.get("IS_MATERIALIZED_VIEW")
+        ):
             continue
         if _value_is_true(row.get("IS_EXTERNAL")):
             continue
@@ -756,7 +808,11 @@ def fetch_yaml_names_in_stage(
     if stage.lower().startswith("volume:user://"):
         volume_body = stage[len("volume:") :]
         # Normalize relative directory
-        relative = volume_body[len("user://") :] if volume_body.startswith("user://") else volume_body
+        relative = (
+            volume_body[len("user://") :]
+            if volume_body.startswith("user://")
+            else volume_body
+        )
         relative = relative.lstrip("~/")
         relative = relative.strip("/")
 
@@ -842,7 +898,9 @@ def create_table_in_schema(
     table_fqn: str,
     columns_schema: Dict[str, str],
 ) -> bool:
-    fields = ", ".join(f"{_quote_identifier(name)} {dtype}" for name, dtype in columns_schema.items())
+    fields = ", ".join(
+        f"{_quote_identifier(name)} {dtype}" for name, dtype in columns_schema.items()
+    )
     query = f"CREATE TABLE IF NOT EXISTS {table_fqn} ({fields})"
     try:
         session.sql(query).collect()

semantic_model_generator/clickzetta_utils/env_vars.py

@@ -20,7 +20,9 @@ _CONFIG_PATHS = [
 _ACTIVE_CONFIG_PATH: Optional[str] = None
 
 
-def _load_config_from_file() -> Tuple[Optional[Dict[str, str]], Dict[str, Dict[str, str]]]:
+def _load_config_from_file() -> (
+    Tuple[Optional[Dict[str, str]], Dict[str, Dict[str, str]]]
+):
     global _ACTIVE_CONFIG_PATH
     _ACTIVE_CONFIG_PATH = None
     for path in _CONFIG_PATHS:
@@ -91,7 +93,10 @@ def _deep_lookup(mapping: Any, key: str) -> Optional[Any]:
         if isinstance(current, dict):
             for candidate_key, candidate_value in current.items():
                 candidate_key_str = str(candidate_key).lower()
-                if candidate_key_str == normalized_key and candidate_value not in (None, ""):
+                if candidate_key_str == normalized_key and candidate_value not in (
+                    None,
+                    "",
+                ):
                     return candidate_value
                 if isinstance(candidate_value, (dict, list)):
                     queue.append(candidate_value)

semantic_model_generator/data_processing/cte_utils.py

@@ -8,10 +8,10 @@ import sqlglot.expressions
 from loguru import logger
 from sqlglot import Dialect
 
-from semantic_model_generator.protos import semantic_model_pb2
 from semantic_model_generator.clickzetta_utils.clickzetta_connector import (
     OBJECT_DATATYPES,
 )
+from semantic_model_generator.protos import semantic_model_pb2
 
 _SQLGLOT_CLICKZETTA_KEY = "".join(["snow", "flake"])
 ClickzettaDialect = Dialect.get_or_raise(_SQLGLOT_CLICKZETTA_KEY)