adjust interface

lib/gpt/qcd/pseudofermion/action/base.py

@@ -52,14 +52,14 @@ def draw(self, fields, rng):
     def gradient(self, fields, dfields):
         raise NotImplementedError()
 
-    def transformed(self, t):
-        return transformed_action(self, t)
+    def transformed(self, t, **args):
+        return transformed_action(self, t, **args)
 
 
 class transformed_action(action_base):
-    def __init__(self, a, t):
+    def __init__(self, a, t, **args):
         self.a = a
-        self.at = differentiable_functional.transformed(a, t)
+        self.at = differentiable_functional.transformed(a, t, **args)
         self.t = t
 
     def __call__(self, fields):

lib/gpt/qcd/pseudofermion/action/exact_one_flavor.py

@@ -159,6 +159,7 @@ def gradient(self, fields, dfields):
             if mu < len(fields) - 1:
                 dS.append(g.qcd.gauge.project.traceless_hermitian(frc[mu]))
             else:
-                raise Exception("not implemented")
+                # not yet implemented
+                dS.append(None)
 
         return dS

tests/qcd/fermion_actions.py

@@ -87,7 +87,7 @@ def mobius(m_plus, m_minus):
 ]
 
 sm = g.qcd.gauge.smear.stout(rho=0.157)
-a_sm = acts[0][0].transformed(sm)
+a_sm = acts[0][0].transformed(sm, indices=[0, 1, 2, 3])
 a_sm.assert_gradient_error(rng, U + [acts[0][2]], U, 1e-3, 5e-7)
 
 for _a in acts: