From de3f87f731bbcd5a889fd00c8624bf37362a27be Mon Sep 17 00:00:00 2001 From: Eric Prestat Date: Sat, 28 Oct 2023 19:00:42 +0200 Subject: [PATCH] Fix test cuda --- hyperspy/tests/signals/test_cuda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hyperspy/tests/signals/test_cuda.py b/hyperspy/tests/signals/test_cuda.py index 8679cfb2b1..4803e123ba 100644 --- 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)