From bb223a89ddb01322b17d81e1a1beb7377f1a61bb Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Wed, 27 Jun 2018 17:45:27 +0200
Subject: [PATCH] Fixup

---
 nifty5/operators/domain_distributor.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/nifty5/operators/domain_distributor.py b/nifty5/operators/domain_distributor.py
index 4596284f8..a555196b1 100644
--- a/nifty5/operators/domain_distributor.py
+++ b/nifty5/operators/domain_distributor.py
@@ -5,12 +5,12 @@ from .. import dobj
 from ..domain_tuple import DomainTuple
 from .linear_operator import LinearOperator
 
-if dobj.ntask > 1:
-    raise NotImplementedError('UpProj class does not support MPI.')
-
 
 class DomainDistributor(LinearOperator):
     def __init__(self, target, axis):
+        # TODO Replace this by a DiagonalOperator
+        if dobj.ntask > 1:
+            raise NotImplementedError('UpProj class does not support MPI.')
         assert len(target) == 2
         assert axis in [0, 1]
 
-- 
GitLab