Start fixing grid

src/eko/evolution_operator/grid.py

@@ -123,8 +123,7 @@ def get_threshold_operators(self, path: List[Segment]) -> List[Operator]:
             ome = OperatorMatrixElement(
                 self.config,
                 self.managers,
-                seg.nf - shift + 3,
-                seg.target,
+                seg,
                 is_downward,
                 np.log(kthr),
                 self.config["HQ"] == "MSBAR",

src/eko/evolution_operator/operator_matrix_element.py

@@ -182,12 +182,8 @@ class OperatorMatrixElement(Operator):
         configuration
     managers : dict
         managers
-    nf: int
-        number of active flavor below threshold
-    q2: float
-        matching scale
-    is_backward: bool
-        True for backward evolution
+    segment: Segment
+        path segment
     L: float
         :math:`\ln(\mu_F^2 / m_h^2)`
     is_msbar: bool

tests/eko/evolution_operator/test_grid.py

@@ -44,33 +44,26 @@ def test_compute_mu2grid(theory_ffns, operator_card, tmp_path):
     opgrid = legacy.Runner(
         theory_ffns(3), operator_card, path=tmp_path / "eko.tar"
     ).op_grid
-    # q2 has not be precomputed - but should work nevertheless
-    opgrid.compute(3)
-    # we can also pass a single number
     opg = opgrid.compute()
     assert len(opg) == len(mugrid)
     assert all(k in op for k in ["operator", "error"] for op in opg.values())
-    opg = opgrid.compute(3)
-    assert len(opg) == 1
-    assert all(k in op for k in ["operator", "error"] for op in opg.values())
 
 
 def test_grid_computation_VFNS(theory_card, operator_card, tmp_path):
     """Checks that the grid can be computed"""
+    mugrid = [(3, 4), (5, 5), (5, 4)]
+    operator_card.mugrid = mugrid
     opgrid = legacy.Runner(
         theory_card, operator_card, path=tmp_path / "eko.tar"
     ).op_grid
-    qgrid_check = [3, 5, 200**2]
-    operators = opgrid.compute(qgrid_check)
-    assert len(operators) == len(qgrid_check)
+    operators = opgrid.compute()
+    assert len(operators) == len(mugrid)
 
 
 def test_mod_expanded(theory_card, theory_ffns, operator_card, tmp_path: pathlib.Path):
+    mugrid = [(3, 4)]
+    operator_card.mugrid = mugrid
     operator_card.configs.scvar_method = eko.io.types.ScaleVariationsMethod.EXPANDED
-    theory_update = {
-        "order": (1, 0),
-        "ModSV": "expanded",
-    }
     epsilon = 1e-1
     path = tmp_path / "eko.tar"
     for is_ffns, nf0 in zip([False, True], [5, 3]):
@@ -82,11 +75,11 @@ def test_mod_expanded(theory_card, theory_ffns, operator_card, tmp_path: pathlib
         theory.heavy.num_flavs_init = nf0
         path.unlink(missing_ok=True)
         opgrid = legacy.Runner(theory, operator_card, path=path).op_grid
-        opg = opgrid.compute(3)
-        theory_update["XIF"] = 1.0 + epsilon
+        opg = opgrid.compute()
+        theory.xif = 1.0 + epsilon
         path.unlink(missing_ok=True)
         sv_opgrid = legacy.Runner(theory, operator_card, path=path).op_grid
-        sv_opg = sv_opgrid.compute(3)
+        sv_opg = sv_opgrid.compute()
         np.testing.assert_allclose(
-            opg[3]["operator"], sv_opg[3]["operator"], atol=2.5 * epsilon
+            opg[(9, 4)]["operator"], sv_opg[(9, 4)]["operator"], atol=2.5 * epsilon
         )