Merge pull request #121 from coneoproject/citations-module

* citations-module: Fix some flake8 errors Migrate citations import to firedrake_citations
coneoproject · Nov 10, 2017 · 5bb1b30 · 5bb1b30
2 parents 029d2af + a182af3
commit 5bb1b30
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/coffee/citations.py b/coffee/citations.py
@@ -53,7 +53,7 @@ def update_citations(params):
 
     # Firedrake
     try:
-        from firedrake import Citations
+        from firedrake_citations import Citations
         rewrite = params.get('rewrite', 0)
         nozeros = params.get('dead_ops_elimination', False)
         align_pad = params.get('align_pad', False)

diff --git a/coffee/scheduler.py b/coffee/scheduler.py
@@ -520,7 +520,7 @@ def _track_nz_expr(self, node, nz_syms, nest):
                 try:
                     # Am I tracking the loop with iteration variable == /r/ ?
                     loop = [l for l, p in nest if l.dim == r][0]
-                except:
+                except IndexError:
                     # No, so I just assume it covers the entire non zero-valued region
                     itspace.append([(r, nz_b) for nz_b in nz_bs])
                     continue

diff --git a/coffee/vectorizer.py b/coffee/vectorizer.py
@@ -491,7 +491,7 @@ def __init__(self, tensor_size):
 
         def get_reg(self):
             if len(self.var) == 0:
-                l = self.ntot * 2
+                l = self.ntot * 2  # noqa: E741
                 self.var += [self.i["reg"](v) for v in range(self.ntot, l)]
                 self.ntot = l
             return self.var.pop(0)