feat: lazy import numpy like pandas (#4653)

Author: eakmanrq

examples/sushi/helper.py

@@ -2,7 +2,7 @@
 import typing as t
 from datetime import datetime, timedelta
 
-import numpy as np
+import numpy as np  # noqa: TID253
 
 
 def set_seed(dt: datetime) -> None:

examples/sushi/models/items.py

@@ -2,7 +2,7 @@
 import typing as t
 from datetime import datetime
 
-import numpy as np
+import numpy as np  # noqa: TID253
 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

@@ -2,7 +2,7 @@
 import typing as t
 from datetime import datetime
 
-import numpy as np
+import numpy as np  # noqa: TID253
 import pandas as pd  # noqa: TID253
 from helper import iter_dates  # type: ignore
 from sqlglot import exp

examples/sushi/models/raw_marketing.py

@@ -2,7 +2,7 @@
 import typing as t
 from datetime import datetime
 
-import numpy as np
+import numpy as np  # noqa: TID253
 import pandas as pd  # noqa: TID253
 from sqlglot import exp
 

examples/wursthall/models/db/order_f.py

@@ -2,7 +2,7 @@
 import typing as t
 from datetime import datetime
 
-import numpy as np
+import numpy as np  # noqa: TID253
 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/order_item_details.py

@@ -3,7 +3,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 
-import numpy as np
+import numpy as np  # noqa: TID253
 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/shared.py

@@ -4,7 +4,7 @@
 import typing as t
 from datetime import date, timedelta
 
-import numpy as np
+import numpy as np  # noqa: TID253
 from faker import Faker
 
 SEED = 99999999

pyproject.toml

@@ -275,5 +275,6 @@ extend-select = ["TID"]
 
 [tool.ruff.lint.flake8-tidy-imports]
 banned-module-level-imports = [
-  "pandas",
+    "pandas",
+    "numpy",
 ]

sqlmesh/core/console.py

@@ -8,7 +8,6 @@
 import logging
 import textwrap
 from pathlib import Path
-import numpy as np
 from hyperscript import h
 from rich.console import Console as RichConsole
 from rich.live import Live
@@ -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
+    import numpy as np
 
     # Convert array-like objects to list for consistent comparison
     def _normalize(val: t.Any) -> t.Any:

sqlmesh/core/engine_adapter/mssql.py

@@ -4,7 +4,6 @@
 
 import typing as t
 
-import numpy as np
 from sqlglot import exp
 
 from sqlmesh.core.dialect import to_schema
@@ -215,6 +214,7 @@ def _df_to_source_queries(
         target_table: TableName,
     ) -> t.List[SourceQuery]:
         import pandas as pd
+        import numpy as np
 
         assert isinstance(df, pd.DataFrame)
         temp_table = self._get_temp_table(target_table or "pandas")