Add mypy

src/eko/evolution_operator/__init__.py

@@ -579,6 +579,9 @@ def quad_ker_qed(
     return ker
 
 
+OPMEMBERS = Dict[Tuple[int, int], OpMember]
+
+
 class Operator(sv.ModeMixin):
     """Internal representation of a single EKO.
 
@@ -618,7 +621,7 @@ def __init__(
         # TODO make 'cut' external parameter?
         self._mellin_cut = mellin_cut
         self.is_threshold = is_threshold
-        self.op_members: Dict[Tuple[int, int], OpMember] = {}
+        self.op_members: OPMEMBERS = {}
         self.order = tuple(config["order"])
         self.alphaem_running = self.managers["couplings"].alphaem_running
         if self.log_label == "Evolution":

src/eko/evolution_operator/grid.py

@@ -5,7 +5,6 @@
 
 """
 import logging
-from dataclasses import astuple
 from typing import Any, Dict, List, Optional
 
 import numpy as np
@@ -17,9 +16,9 @@
 from ..interpolation import InterpolatorDispatcher
 from ..io.runcards import Configs, Debug
 from ..io.types import EvolutionPoint as EPoint
-from ..io.types import Order
+from ..io.types import Order, SquaredScale
 from ..matchings import Atlas, Segment, flavor_shift, is_downward_path
-from . import Operator, flavors, matching_condition, physical
+from . import OPMEMBERS, Operator, flavors, matching_condition, physical
 from .operator_matrix_element import OperatorMatrixElement
 
 logger = logging.getLogger(__name__)
@@ -95,8 +94,8 @@ def __init__(
             couplings=couplings,
             interpol_dispatcher=interpol_dispatcher,
         )
-        self._threshold_operators = {}
-        self._matching_operators = {}
+        self._threshold_operators: Dict[Segment, Operator] = {}
+        self._matching_operators: Dict[SquaredScale, OPMEMBERS] = {}
 
     def get_threshold_operators(self, path: List[Segment]) -> List[Operator]:
         """Generate the threshold operators.
@@ -118,7 +117,7 @@ def get_threshold_operators(self, path: List[Segment]) -> List[Operator]:
         is_downward = is_downward_path(path)
         shift = flavor_shift(is_downward)
         for seg in path[:-1]:
-            new_op_key = astuple(seg)
+            new_op_key = seg
             kthr = self.config["thresholds_ratios"][seg.nf - shift]
             ome = OperatorMatrixElement(
                 self.config,

src/eko/msbar_masses.py

@@ -395,7 +395,7 @@ def sc(thr_masses):
     heavy_quarks = quark_names[3:]
     hq_idxs = np.arange(0, 3)
     if nf_ref > 4:
-        heavy_quarks = str(reversed(heavy_quarks))
+        heavy_quarks = "".join([e for e in reversed(heavy_quarks)])
         hq_idxs = reversed(hq_idxs)
 
     # loop on heavy quarks and compute the msbar masses