Supported options are now a class attribute.

gext/descriptors.py

@@ -7,6 +7,8 @@ class Distance:
 
     """Distance matrix descriptors."""
 
+    supported_options = {}
+
     def __init__(self, **kwargs):
         self.set_options(**kwargs)
 
@@ -24,6 +26,8 @@ class Coulomb(Distance):
 
     """Coulomb matrix descriptors."""
 
+    supported_options = {}
+
     def compute(self, coords: np.ndarray) -> np.ndarray:
         """Compute the Coulomb matrix as a descriptor."""
         return 1.0/super().compute(coords)

gext/fitting.py

@@ -8,8 +8,9 @@ class AbstractFitting(abc.ABC):
 
     """Base class for fitting schemes."""
 
+    supported_options = {}
+
     def __init__(self, **kwargs):
-        self.supported_options = {}
         self.set_options(**kwargs)
 
     @abc.abstractmethod
@@ -26,7 +27,6 @@ class AbstractFitting(abc.ABC):
             if option not in self.options:
                 self.options[option] = default_value
 
-
     @abc.abstractmethod
     def fit(self, vectors: List[np.ndarray], target:np.ndarray):
         """Base method for computing new fitting coefficients."""
@@ -45,10 +45,11 @@ class LeastSquare(AbstractFitting):
 
     """Simple least square minimization fitting."""
 
-    def __init__(self, **kwargs):
-        self.supported_options = {
+    supported_options = {
         "regularization": 0.0,
     }
+
+    def __init__(self, **kwargs):
         super().__init__(**kwargs)
 
     def set_options(self, **kwargs):
@@ -70,10 +71,11 @@ class QuasiTimeReversible(AbstractFitting):
 
     """Quasi time reversible fitting scheme. Not yet implemented."""
 
-    def __init__(self, **kwargs):
-        self.supported_options = {
+    supported_options = {
         "regularization": 0.0,
     }
+
+    def __init__(self, **kwargs):
         super().__init__(**kwargs)
 
     def set_options(self, **kwargs):

gext/main.py

@@ -14,9 +14,7 @@ class Extrapolator:
     it requires the number of electrons, the number of basis functions
     and the number of atoms of the molecule."""
 
-    def __init__(self, nelectrons: int, nbasis: int, natoms: int, **kwargs):
-
-        self.supported_options = {
+    supported_options = {
         "verbose": False,
         "nsteps": 6,
         "descriptor": "distance",
@@ -25,6 +23,8 @@ class Extrapolator:
         "store_overlap": True,
     }
 
+    def __init__(self, nelectrons: int, nbasis: int, natoms: int, **kwargs):
+
         self.nelectrons = nelectrons
         self.nbasis = nbasis
         self.natoms = natoms