Merge pull request #6 from szapp/proxfunKwargs

Pass kwargs to external proximal operator functions
ganguli-lab · May 11, 2019 · f39f633 · f39f633
2 parents 3d3e74d + daa0c83
commit f39f633
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/proxalgs/core.py b/proxalgs/core.py
@@ -85,7 +85,7 @@ def add_regularizer(self, proxfun, **kwargs):
 
         # if proxfun is a function, add it as its own proximal operator
         elif hasattr(proxfun, '__call__'):
-            self.objectives.append(lambda theta, rho: proxfun(theta.copy(), float(rho)))
+            self.objectives.append(lambda theta, rho: proxfun(theta.copy(), float(rho), **kwargs))
 
         # type of proxfun must be a string or a function
         else: