Fix CI failures

inducer · Aug 4, 2024 · 9525a18 · 9525a18
1 parent 413bc96
commit 9525a18
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pytential/qbx/distributed.py b/pytential/qbx/distributed.py
@@ -222,7 +222,7 @@ def compute_local_geometry_data(
         data is generated.
     :arg boxes_time: Predicated cost of each box. Used by partitioning to improve
         load balancing.
-    :arg traversal_builder: Used to construct local tranversal.
+    :arg traversal_builder: Used to construct local traversal.
     """
     queue = actx.queue
 

diff --git a/test/test_distributed.py b/test/test_distributed.py
@@ -4,7 +4,8 @@
 import pyopencl.clmath
 from pyopencl.tools import (  # noqa
         pytest_generate_tests_for_pyopencl as pytest_generate_tests)
-from arraycontext import PyOpenCLArrayContext, flatten, unflatten
+from arraycontext import flatten, unflatten
+from meshmode.array_context import PyOpenCLArrayContext
 
 from meshmode.mesh.generation import make_curve_mesh, ellipse
 from sumpy.visualization import FieldPlotter