quick workaround to have __localop in cw4heat

heat/cw4heat/__init__.py

@@ -135,8 +135,8 @@ def fini(self):
                 pass
 
         c = MPI.COMM_WORLD
-        if c.size <= 1:
-            raise Exception("At least 2 ranks required for cw4heat")
+        # if c.size <= 1:
+        #    raise Exception("At least 2 ranks required for cw4heat")
         _runner = MPIRunner(Distributor(c), c)
         if doStart:
             _runner.distributor.start(initImpl=_setComm)
@@ -392,6 +392,14 @@ def __getattr__(self, attr):
     )
 
 
+def __local_op_normalized(a, f):
+    return impl.core._operations.__local_op(f, a)
+
+
+def __local_op(*args, **kwargs):
+    return DDParray(_submit("__local_op_normalized", args, kwargs))
+
+
 # Here we define data types and constants
 for attr in aa_datatypes + aa_constants:
     if hasattr(impl, attr):