feat: remove global pandas import to improve import speed

Author: eakmanrq

examples/ibis/models/ibis_full_model_python.py

@@ -2,7 +2,7 @@
 from datetime import datetime
 
 import ibis  # type: ignore
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from constants import DB_PATH  # type: ignore
 from sqlglot import exp
 

examples/sushi/models/items.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 
 import numpy as np
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from helper import iter_dates  # type: ignore
 from sqlglot.expressions import to_column
 

examples/sushi/models/order_items.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 
 import numpy as np
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from helper import iter_dates  # type: ignore
 from sqlglot import exp
 from sqlglot.expressions import to_column

examples/sushi/models/orders.py

@@ -2,7 +2,7 @@
 import typing as t
 from datetime import datetime, timedelta
 
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from helper import iter_dates  # type: ignore
 
 from sqlmesh import ExecutionContext, model

examples/sushi/models/raw_marketing.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 
 import numpy as np
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from sqlglot import exp
 
 from sqlmesh import ExecutionContext, model

examples/wursthall/models/db/order_f.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 
 import numpy as np
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from models.src.shared import DATA_START_DATE_STR, set_seed  # type: ignore
 from sqlglot import parse_one
 

examples/wursthall/models/src/customer_details.py

@@ -3,7 +3,7 @@
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from faker import Faker
 from models.src.shared import DATA_START_DATE_STR, iter_dates, set_seed  # type: ignore
 

examples/wursthall/models/src/order_item_details.py

@@ -4,7 +4,7 @@
 from datetime import datetime
 
 import numpy as np
-import pandas as pd
+import pandas as pd  # noqa: TID253
 from faker import Faker
 from models.src.shared import DATA_START_DATE_STR, iter_dates, set_seed  # type: ignore
 from sqlglot import parse_one

pyproject.toml

@@ -271,3 +271,9 @@ select = [
     "F401",
     "RET505",
 ]
+extend-select = ["TID"]
+
+[tool.ruff.lint.flake8-tidy-imports]
+banned-module-level-imports = [
+  "pandas",
+]

sqlmesh/core/console.py

@@ -8,7 +8,6 @@
 import logging
 import textwrap
 from pathlib import Path
-import pandas as pd
 import numpy as np
 from hyperscript import h
 from rich.console import Console as RichConsole
@@ -2497,6 +2496,7 @@ def show_linter_violations(
 
 def _cells_match(x: t.Any, y: t.Any) -> bool:
     """Helper function to compare two cells and returns true if they're equal, handling array objects."""
+    import pandas as pd
 
     # Convert array-like objects to list for consistent comparison
     def _normalize(val: t.Any) -> t.Any: