Go again

examples/nemo/scripts/passthrough.py

@@ -37,16 +37,19 @@
 ''' Process Nemo code with PSyclone but don't do any changes. This file is only
 needed to provide a FILES_TO_SKIP list. '''
 from utils import PASSTHROUGH_ISSUES
+# TODO Remove
 from psyclone.psyir.transformations import ScalarizationTrans
+from psyclone.psyir.nodes import Routine, Loop
 
 # List of all files that psyclone will skip processing
 FILES_TO_SKIP = PASSTHROUGH_ISSUES
 
 
-def trans(_):
+def trans(psyir):
     ''' Don't do any changes. '''
     # TODO REMOVE
-    scalartrans = ScalarizationTrans()
-    for loop in subroutine.walk(Loop):
-        scalartrans.apply(loop)
+    for subroutine in psyir.walk(Routine):
+        scalartrans = ScalarizationTrans()
+        for loop in subroutine.walk(Loop):
+            scalartrans.apply(loop)