Fix flake8

tberlok · Feb 29, 2024 · 0dca5a2 · 0dca5a2
1 parent ffee8a7
commit 0dca5a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/cuda-gpu/test_gpu_binary_tree.py b/tests/cuda-gpu/test_gpu_binary_tree.py
@@ -4,7 +4,7 @@
 @pytest.mark.xfail(raises=RuntimeError)
 def test_gpu_binary_tree():
     try:
-        import cupy
+        import cupy as cp
         import numba
     except ImportError:
         msg = ("Import of cupy and numba failed "