Fix: Make sure to commit a transaction when fetching a dataframe using Pandas API (#2732)

izeigerman · web-flow · commit 9c5b985f1bb9 · 2024-06-04T13:12:00.000-07:00
diff --git a/sqlmesh/core/engine_adapter/mixins.py b/sqlmesh/core/engine_adapter/mixins.py
@@ -61,7 +61,7 @@ def _fetch_native_df(
             else query
         )
         logger.debug(f"Executing SQL:\n{sql}")
-        with catch_warnings():
+        with catch_warnings(), self.transaction():
             filterwarnings(
                 "ignore",
                 category=UserWarning,

Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@ def _fetch_native_df(`
`61`	`61`	`else query`
`62`	`62`	`)`
`63`	`63`	`logger.debug(f"Executing SQL:\n{sql}")`
`64`		`- with catch_warnings():`
	`64`	`+ with catch_warnings(), self.transaction():`
`65`	`65`	`filterwarnings(`
`66`	`66`	`"ignore",`
`67`	`67`	`category=UserWarning,`