fix accumulator device error

pytorch · Sep 14, 2024 · db6392d · db6392d
1 parent 1c0b1a4
commit db6392d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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}"