From d1bb2b7d00a58bbe08387f203235a94a1eb7647c Mon Sep 17 00:00:00 2001
From: Philippe Laferriere <plafer@protonmail.com>
Date: Tue, 12 Mar 2024 14:49:34 -0400
Subject: [PATCH] Remove `Air::get_lagrange_kernel_transition_constraints`

---
 air/src/air/mod.rs                          |  7 -------
 prover/src/constraints/evaluator/default.rs |  3 ++-
 verifier/src/evaluator.rs                   | 10 +++++-----
 3 files changed, 7 insertions(+), 13 deletions(-)

diff --git a/air/src/air/mod.rs b/air/src/air/mod.rs
index e98a36e74..1bd9f07a1 100644
--- a/air/src/air/mod.rs
+++ b/air/src/air/mod.rs
@@ -394,13 +394,6 @@ pub trait Air: Send + Sync {
         TransitionConstraints::new(self.context(), composition_coefficients)
     }
 
-    fn get_lagrange_kernel_transition_constraints<E: FieldElement<BaseField = Self::BaseField>>(
-        &self,
-        lagrange_constraint_coefficients: Vec<E>,
-    ) -> LagrangeKernelTransitionConstraints<E> {
-        LagrangeKernelTransitionConstraints::new(self.context(), lagrange_constraint_coefficients)
-    }
-
     /// Convert assertions returned from [get_assertions()](Air::get_assertions) and
     /// [get_aux_assertions()](Air::get_aux_assertions) methods into boundary constraints.
     ///
diff --git a/prover/src/constraints/evaluator/default.rs b/prover/src/constraints/evaluator/default.rs
index 7e3e6205d..d6df62241 100644
--- a/prover/src/constraints/evaluator/default.rs
+++ b/prover/src/constraints/evaluator/default.rs
@@ -154,7 +154,8 @@ where
 
         let lagrange_kernel_transition_constraints =
             air.context().lagrange_kernel_aux_column_idx().map(|_| {
-                air.get_lagrange_kernel_transition_constraints(
+                LagrangeKernelTransitionConstraints::new(
+                    air.context(),
                     composition_coefficients.lagrange_kernel_transition,
                 )
             });
diff --git a/verifier/src/evaluator.rs b/verifier/src/evaluator.rs
index bec587359..5382b52af 100644
--- a/verifier/src/evaluator.rs
+++ b/verifier/src/evaluator.rs
@@ -5,7 +5,7 @@
 
 use air::{
     Air, AuxTraceRandElements, ConstraintCompositionCoefficients, EvaluationFrame,
-    LagrangeKernelEvaluationFrame,
+    LagrangeKernelEvaluationFrame, LagrangeKernelTransitionConstraints,
 };
 use math::{polynom, FieldElement};
 use utils::collections::*;
@@ -92,10 +92,10 @@ pub fn evaluate_constraints<A: Air, E: FieldElement<BaseField = A::BaseField>>(
     // 3 ----- evaluate Lagrange kernel transition constraints ------------------------------------
 
     if let Some(lagrange_kernel_column_frame) = lagrange_kernel_column_frame {
-        let lagrange_kernel_transition_constraints = air
-            .get_lagrange_kernel_transition_constraints(
-                composition_coefficients.lagrange_kernel_transition,
-            );
+        let lagrange_kernel_transition_constraints = LagrangeKernelTransitionConstraints::new(
+            air.context(),
+            composition_coefficients.lagrange_kernel_transition,
+        );
         let lagrange_kernel_column_rand_elements =
             air.lagrange_kernel_rand_elements(aux_rand_elements.get_segment_elements(0));