PR cleanup

ecmwf-ifs · Jun 11, 2024 · cf8c8e5 · cf8c8e5
1 parent 6c6e661
commit cf8c8e5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/loki/analyse/analyse_dataflow.py b/loki/analyse/analyse_dataflow.py
@@ -275,8 +275,7 @@ def visit_Import(self, o, **kwargs):
 
     def visit_VariableDeclaration(self, o, **kwargs):
         defines = self._symbols_from_expr(o.symbols, condition=lambda v: v.type.initial is not None)
-        arrays = [a for a in o.symbols if isinstance(a, Array)]
-        uses = set(v for a in arrays for v in FindVariables().visit(a.dimensions))
+        uses = {v for a in o.symbols if isinstance(a, Array) for v in self._symbols_from_expr(a.dimensions)}
         return self.visit_Node(o, defines_symbols=defines, uses_symbols=uses, **kwargs)
 
 

diff --git a/loki/transformations/tests/test_data_offload.py b/loki/transformations/tests/test_data_offload.py
@@ -433,8 +433,6 @@ def test_global_variable_analysis(frontend, key, config, global_variable_analysi
         if item == 'global_var_analysis_data_mod#some_type':
             continue
         for trafo_data_key, trafo_data_value in item.trafo_data[key].items():
-            print(item)
-            print(trafo_data_value)
             assert (
                 sorted(
                     tuple(str(vv) for vv in v) if isinstance(v, tuple) else str(v)