From db6392dc9861ea8b577a578a0cf01381505eb48b Mon Sep 17 00:00:00 2001 From: Kazuki Adachi Date: Sat, 14 Sep 2024 22:34:00 +0900 Subject: [PATCH] fix accumulator device error --- tests/ignite/metrics/test_hsic.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ignite/metrics/test_hsic.py b/tests/ignite/metrics/test_hsic.py index a8fc6b404b9..5b20a37bb0a 100644 --- a/tests/ignite/metrics/test_hsic.py +++ b/tests/ignite/metrics/test_hsic.py @@ -173,12 +173,12 @@ def test_accumulator_device(self): for metric_device in metric_devices: hsic = HSIC(device=metric_device) - for dev in (hsic._device, hsic._sum_of_hsic): + for dev in (hsic._device, hsic._sum_of_hsic.device): assert dev == metric_device, f"{type(dev)}:{dev} vs {type(metric_device)}:{metric_device}" x = torch.zeros(10, 10).float() y = torch.ones(10, 10).float() hsic.update((x, y)) - for dev in (hsic._device, hsic._sum_of_hsic): + for dev in (hsic._device, hsic._sum_of_hsic.device): assert dev == metric_device, f"{type(dev)}:{dev} vs {type(metric_device)}:{metric_device}"