Refactored out dcite call to avoid defining __init__.

Author: cgranade

src/qinfer/test_models.py

@@ -55,7 +55,7 @@
 
 import numpy as np
 
-from .utils import binomial_pdf
+from .utils import binomial_pdf, decorate_init
 from .abstract_model import FiniteOutcomeModel, DifferentiableModel
 from ._due import due, Doi
 
@@ -212,6 +212,13 @@ def score(self, outcomes, modelparams, expparams, return_L=False):
         else:
             return q
 
+@decorate_init(
+    due.dcite(
+        Doi('10.1088/1367-2630/14/10/103013'),
+        description='Robust online Hamiltonian learning',
+        tags=['implementation']
+    )
+)
 class UnknownT2Model(FiniteOutcomeModel):
     """
     Describes the free evolution of a single qubit prepared in the
@@ -223,11 +230,6 @@ class UnknownT2Model(FiniteOutcomeModel):
     :modelparam T2_inv: The decoherence strength :math:`T_2^{-1}`.
     :scalar-expparam float: The evolution time :math:`t`.
     """
-    @due.dcite(
-        Doi('10.1088/1367-2630/14/10/103013'),
-        description='Robust online Hamiltonian learning',
-        tags=['implementation']
-    )
 
     @property
     def n_modelparams(self): return 2

src/qinfer/utils.py

@@ -606,6 +606,20 @@ def sqrtm_psd(A, est_error=True, check_finite=True):
     else:
         return A_sqrt
 
+def decorate_init(init_decorator):
+    """
+    Given a class definition and a decorator that acts on methods,
+    applies that decorator to the class' __init__ method.
+    Useful for decorating __init__ while still allowing __init__ to be
+    inherited.
+    """
+
+    def class_decorator(cls):
+        cls.__init__ = init_decorator(cls.__init__)
+        return cls
+
+    return class_decorator
+
 #==============================================================================
 #Test Code
 if __name__ == "__main__":