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