diff --git a/pytential/qbx/distributed.py b/pytential/qbx/distributed.py index 309e43570..9cca6dddb 100644 --- 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 index 5b2f38393..e69c100ec 100644 --- 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