From 4fa7994d4992238836628235fff50492f9a33873 Mon Sep 17 00:00:00 2001
From: Dominic Kempf <dominic.r.kempf@gmail.com>
Date: Fri, 28 Aug 2015 15:59:28 +0200
Subject: [PATCH] Fixup

---
 python/dune/perftool/validity.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/python/dune/perftool/validity.py b/python/dune/perftool/validity.py
index 134ce473..b88b6881 100644
--- a/python/dune/perftool/validity.py
+++ b/python/dune/perftool/validity.py
@@ -1,3 +1,4 @@
+from ufl.classes import Sum
 from ufl.algorithms import MultiFunction
 
 
@@ -37,7 +38,7 @@ class UFLValidityChecker(MultiFunction):
     def sum(self, o, topsum):
         if topsum:
             for op in o.operands():
-                if not isinstance(op, ufl.classes.Sum):
+                if not isinstance(op, Sum):
                     self.sane = len(MultiFunction.__call__(self, op, True)) == self.rank
         else:
             # If this is sum is not part of the topsum, we treat it as all other expressions
@@ -47,7 +48,7 @@ class UFLValidityChecker(MultiFunction):
         assert len(o.operands()) == 2
         if topsum:
             op = o.operands()[0]
-            if not isinstance(op, ufl.classes.Sum):
+            if not isinstance(op, Sum):
                 self.sane = len(MultiFunction.__call__(self, op, True)) == self.rank
         else:
             return self.expr(o, topsum)
-- 
GitLab