misc: Replace try-except with if-else

devitocodes · Sep 16, 2024 · de56705 · de56705
1 parent 925037b
commit de56705
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/devito/ir/equations/algorithms.py b/devito/ir/equations/algorithms.py
@@ -264,9 +264,9 @@ def _(d, mapper, rebuilt, sregistry):
 
     idim0 = d.implicit_dimension
     if idim0 is not None:
-        try:
+        if idim0 in rebuilt:
             idim1 = rebuilt[idim0]
-        except KeyError:
+        else:
             iname = sregistry.make_name(prefix='n')
             rebuilt[idim0] = idim1 = idim0._rebuild(name=iname)
 
@@ -275,9 +275,9 @@ def _(d, mapper, rebuilt, sregistry):
                         'halo': None,
                         'padding': None})
 
-    try:
+    if d.functions in rebuilt:
         functions = rebuilt[d.functions]
-    except KeyError:
+    else:
         fkwargs.update({'name': sregistry.make_name(prefix=d.functions.name),
                         'function': None})