Lint

GridTools · Sep 19, 2024 · 0a376c0 · 0a376c0
1 parent 8fb7452
commit 0a376c0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/src/gt4py/cartesian/frontend/gtscript_frontend.py b/src/gt4py/cartesian/frontend/gtscript_frontend.py
@@ -1440,20 +1440,20 @@ def visit_Assign(self, node: ast.Assign) -> list:
                         loc=nodes.Location.from_ast_node(t),
                     )
                 # Case of K-offset
-                if (
-                    len(spatial_offset) == 3
-                    and spatial_offset[2] != 0
-                ):
-                    if (self.iteration_order == nodes.IterationOrder.PARALLEL):
+                if len(spatial_offset) == 3 and spatial_offset[2] != 0:
+                    if self.iteration_order == nodes.IterationOrder.PARALLEL:
                         raise GTScriptSyntaxError(
                             message="Assignment to non-zero offsets in K is not available in PARALLEL. Choose FORWARD or BACKWARD.",
-                            loc=nodes.Location.from_ast_node(t))
+                            loc=nodes.Location.from_ast_node(t),
+                        )
                     if self.backend_name == "gt:gpu":
                         import cupy as cp
+
                         if cp.cuda.get_local_runtime_version() < 12000:
                             raise GTScriptSyntaxError(
                                 message="Assignment to non-zero offsets in K is not available in gt:gpu for CUDA<12. Please update CUDA.",
-                                loc=nodes.Location.from_ast_node(t))
+                                loc=nodes.Location.from_ast_node(t),
+                            )
 
             if not self._is_known(name):
                 if name in self.temp_decls:

diff --git a/tests/cartesian_tests/integration_tests/multi_feature_tests/test_code_generation.py b/tests/cartesian_tests/integration_tests/multi_feature_tests/test_code_generation.py
@@ -661,6 +661,7 @@ def test_K_offset_write_conditional(backend):
         pytest.skip("Cuda backend is not capable of K offset write")
     if backend == "gt:gpu":
         import cupy as cp
+
         if cp.cuda.get_local_runtime_version() < 12000:
             pytest.skip(
                 "gt:gpu backend with CUDA 11 is not capable of K offset write, update CUDA if possible"