Remove default qed=False from to_flavor_basis_tensor

NNPDF · Oct 12, 2023 · 0c685dd · 0c685dd
1 parent 9803d22
commit 0c685dd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/eko/member.py b/src/eko/member.py
@@ -266,7 +266,7 @@ def operator_multiply(left, right, operation):
                     new_oms[new_key] += operation(l_op, r_op)
         return new_oms
 
-    def to_flavor_basis_tensor(self, qed: bool = False):
+    def to_flavor_basis_tensor(self, qed: bool):
         """Convert the computations into an rank 4 tensor.
 
         A sparse tensor defined with dot-notation (e.g. ``S.g``) is converted

diff --git a/tests/eko/evolution_operator/test_physical.py b/tests/eko/evolution_operator/test_physical.py
@@ -138,7 +138,7 @@ def test_to_flavor_basis_tensor_ss(self):
             ),
             1,
         )
-        vt, _ = a.to_flavor_basis_tensor()
+        vt, _ = a.to_flavor_basis_tensor(False)
         np.testing.assert_allclose(vt[6, :, 6, :], vt[6, :, 5, :])
         np.testing.assert_allclose(vt[6, :, 6, :], vt[5, :, 6, :])
         np.testing.assert_allclose(vt[6, :, 6, :], SS.value[:, :] / (2 * 3))
@@ -158,7 +158,7 @@ def test_to_flavor_basis_tensor_gg(self):
             ),
             1,
         )
-        vt, _ = a.to_flavor_basis_tensor()
+        vt, _ = a.to_flavor_basis_tensor(False)
         np.testing.assert_allclose(vt[6, :, 6, :], 0)
         np.testing.assert_allclose(vt[7, :, 7, :], gg.value[:, :])
         np.testing.assert_allclose(vt[1, :, :, :], 0)
@@ -199,7 +199,7 @@ def test_to_flavor_basis_tensor_gg_qed(self):
             ),
             1,
         )
-        vt, _ = a.to_flavor_basis_tensor()
+        vt, _ = a.to_flavor_basis_tensor(False)
         np.testing.assert_allclose(vt[6, :, 6, :], 0)
         np.testing.assert_allclose(vt[7, :, 7, :], gg.value[:, :])
         np.testing.assert_allclose(vt[1, :, :, :], 0)