fix: increase trino stage limit temporarily (#3118)

* fix: increase trino `stage` limit temporarily

* fix: conditionally set `trino` config

warehouse/metrics_mesh/models/marts/metrics/metrics_v0.sql

@@ -1,11 +1,20 @@
 MODEL (
   name metrics.metrics_v0,
   kind FULL,
+  dialect trino,
   tags (
     'export'
   )
 );
 
+@DEF(MAX_STAGE_OVERRIDE, 550);
+
+-- TODO(jabolo): Remove Trino session logic once #3117 lands
+@IF(
+  @OR(@gateway = 'trino', @gateway = 'local-trino'),
+  SET SESSION query_max_stage_count = @MAX_STAGE_OVERRIDE
+);
+
 WITH unioned_metric_names AS (
   SELECT *
   FROM metrics.int_metric_names_from_artifact
@@ -58,4 +67,10 @@ SELECT
   raw_definition::TEXT,
   definition_ref::TEXT,
   aggregation_function::TEXT
-FROM metrics_v0_no_casting
+FROM metrics_v0_no_casting;
+
+-- TODO(jabolo): Remove Trino session logic once #3117 lands
+@IF(
+  @OR(@gateway = 'trino', @gateway = 'local-trino'),
+  RESET SESSION query_max_stage_count
+);