Fix test cuda

hyperspy · Oct 28, 2023 · de3f87f · de3f87f
1 parent d31b6ea
commit de3f87f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hyperspy/tests/signals/test_cuda.py b/hyperspy/tests/signals/test_cuda.py
@@ -41,7 +41,7 @@ def setup_method(self, method):
     @pytest.mark.parametrize('as_numpy', [True, False, None])
     def test_call_signal(self, as_numpy):
         s = self.s
-        s2 = s(as_numpy=as_numpy)
+        s2 = s._get_current_data(as_numpy=as_numpy)
         if not as_numpy:
             assert isinstance(s2, cp.ndarray)
             s2 = cp.asnumpy(s2)