ruff

inducer · Sep 6, 2024 · 6d507e1 · 6d507e1
1 parent 27e5a19
commit 6d507e1
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py
@@ -105,8 +105,8 @@
     "ArrayOrContainerOrScalarT",
     "ArrayOrContainerT",
     "ArrayT",
-    "CupyArrayContext",
     "CommonSubexpressionTag",
+    "CupyArrayContext",
     "EagerJAXArrayContext",
     "ElementwiseMapKernelTag",
     "NotAnArrayContainerError",

diff --git a/arraycontext/impl/cupy/fake_numpy.py b/arraycontext/impl/cupy/fake_numpy.py
@@ -160,13 +160,13 @@ def linspace(self, *args, **kwargs):
     def zeros_like(self, ary):
         if isinstance(ary, (int, float, complex)):
             # Cupy does not support zeros_like with scalar arguments
-            ary=cp.array(ary)
+            ary = cp.array(ary)
         return rec_map_array_container(cp.zeros_like, ary)
 
     def ones_like(self, ary):
         if isinstance(ary, (int, float, complex)):
             # Cupy does not support ones_like with scalar arguments
-            ary=cp.array(ary)
+            ary = cp.array(ary)
         return rec_map_array_container(cp.ones_like, ary)
 
     def reshape(self, a, newshape, order="C"):

diff --git a/arraycontext/pytest.py b/arraycontext/pytest.py
@@ -243,7 +243,7 @@ def __str__(self):
         name = d["name"].decode("utf-8")
         return f"<CupyArrayContext> on {cupy.cuda.Device()}:{name}"
 
- 
+
 # {{{ _PytestArrayContextFactory
 
 class _NumpyArrayContextForTests(NumpyArrayContext):

diff --git a/test/test_arraycontext.py b/test/test_arraycontext.py
@@ -1298,7 +1298,6 @@ def test_actx_compile_kwargs(actx_factory):
 
     rng = np.random.default_rng()
 
-
     compiled_rhs = actx.compile(scale_and_orthogonalize)
 
     v_x = rng.uniform(size=10)