work towards sparsity in higher dimensions (not point but line)

proxtoolbox/ProxOperators/P_S_real.py

 import numpy as np
 from .proxoperators import ProxOperator
 
+
 class P_S_real(ProxOperator):
     """
     Projection subroutine for projecting onto support constraints
     """
 
-    def __init__(self,config):
+    def __init__(self, config):
         """
         Initialization
         
@@ -17,7 +18,7 @@ class P_S_real(ProxOperator):
         """
         self.support_idx = config['support_idx']
 
-    def work(self,u):
+    def work(self, u):
         """
         Parameters
         ----------
@@ -27,11 +28,7 @@ class P_S_real(ProxOperator):
         -------
         p_S : array_like - the projection 
         """
-        support_idx = self.support_idx -1
+        support_idx = self.support_idx - 1
         p_S = np.zeros(u.shape, dtype=u.dtype)
         p_S[support_idx] = np.real(u[support_idx])
         return p_S
-
-
-
-