bug: fix signature of FirstDerivative in TV

Author: mrava87

pyproximal/proximal/TV.py

@@ -48,13 +48,12 @@ def __init__(self, dims, sigma=1., niter=10, rtol=1e-4, **kwargs):
     def __call__(self, x):
         x = x.reshape(self.dims)
         if self.ndim == 1:
-            derivOp = FirstDerivative(self.dims[0], dims=None, dir=0, edge=False,
+            derivOp = FirstDerivative(dims=self.dims[0], axis=0, edge=False,
                                       dtype=x.dtype, kind="forward")
             dx = derivOp @ x
             y = np.sum(np.abs(dx), axis=0)
         elif self.ndim >= 2:
             y = 0
-            grads = []
             gradOp = Gradient(self.dims, edge=False, dtype=x.dtype, kind="forward")
             grads = gradOp.matvec(x.ravel())
             grads = grads.reshape((self.ndim, ) + self.dims)