fix example plugin

buremba · Feb 8, 2025 · fa4b6e9 · fa4b6e9
1 parent 1430436
commit fa4b6e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/universql/plugins/snow.py b/universql/plugins/snow.py
@@ -45,7 +45,7 @@ def __init__(self, session: "universql.protocol.session.UniverSQLSession"):
         super().__init__(session)
 
     def start_query(self, ast: typing.Optional[List[sqlglot.exp.Expression]], raw_query: str) -> UQuery:
-        return StageTransformer(ast, raw_query)
+        return StageTransformer(self.session, ast, raw_query)
 
 
 # @register()