#2027 fix error in utils.py

stfc · Aug 7, 2024 · 33abf82 · 33abf82
1 parent 65e7ff9
commit 33abf82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/nemo/scripts/utils.py b/examples/nemo/scripts/utils.py
@@ -36,7 +36,7 @@
 ''' Utilities file to parallelise Nemo code. '''
 
 from psyclone.psyir.nodes import (
-    ArrayMixin, Assignment, Loop, Directive, Container, Reference, CodeBlock,
+    Assignment, Loop, Directive, Container, Reference, CodeBlock,
     Call, Return, IfBlock, Routine, IntrinsicCall)
 from psyclone.psyir.symbols import (
     DataSymbol, INTEGER_TYPE, REAL_TYPE, ArrayType, ScalarType,
@@ -194,7 +194,7 @@ def normalise_loops(
                     Reference2ArrayRangeTrans().apply(reference)
                 except TransformationError:
                     pass
-            if isinstance(reference, ArrayMixin):
+            if hasattr(reference, "indices"):
                 # Look at array-index expressions too.
                 for exprn in reference.indices:
                     if (isinstance(exprn, Reference) and