TL: tentative for better static typing

Author: tlunet

qmat/qcoeff/__init__.py

@@ -11,10 +11,14 @@
 - :class:`butcher` : Runge-Kutta based (Butcher tables)
 """
 import numpy as np
+from typing import Type, TypeVar, Dict
 
 from qmat.utils import checkOverriding, storeClass, importAll
 from qmat.lagrange import LagrangeApproximation
 
+T = TypeVar("T")
+
+
 class QGenerator(object):
     """Base abstract class for all :math:`Q`-coefficients generators"""
 
@@ -205,10 +209,10 @@ def errorDahlquist(self, lam, u0, T, nSteps, uNum=None, useEmbeddedWeights=False
         return np.linalg.norm(uNum-uExact, ord=np.inf)
 
 
-Q_GENERATORS = {}
+Q_GENERATORS: Dict[str, QGenerator] = {}
 """Dictionary containing all specialized :class:`QGenerator` classes, with all their aliases"""
 
-def register(cls:QGenerator)->QGenerator:
+def register(cls: Type[T]) -> Type[T]:
     """Class decorator to register a specialized :class:`QGenerator` class in qmat"""
     # Check for correct overriding
     for name in ["nodes", "Q", "weights", "order"]:

qmat/qdelta/__init__.py

@@ -13,9 +13,12 @@
 """
 import inspect
 import numpy as np
+from typing import Type, TypeVar, Dict
 
 from qmat.utils import checkOverriding, storeClass, importAll, checkGenericConstr
 
+T = TypeVar("T")
+
 
 class QDeltaGenerator(object):
     r"""
@@ -154,10 +157,10 @@ def genCoeffs(self, k=None, form="Z2N", dTau=False):
         return out if len(out) > 1 else out[0]
 
 
-QDELTA_GENERATORS = {}
+QDELTA_GENERATORS: Dict[str, QDeltaGenerator] = {}
 """Dictionary containing all specialized :class:`QDeltaGenerator` classes"""
 
-def register(cls:QDeltaGenerator)->QDeltaGenerator:
+def register(cls: Type[T]) -> Type[T]:
     """Class decorator to register a specialized :class:`QDeltaGenerator` class in `qmat`"""
     checkGenericConstr(cls)
     checkOverriding(cls, "computeQDelta", isProperty=False)