fix error in making y

pytorch · Sep 14, 2024 · 5cb0bf1 · 5cb0bf1
1 parent db6392d
commit 5cb0bf1
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
@@ -140,8 +140,8 @@ def test_integration(self, sigma_x: float, sigma_y: float):
         for metric_device in metric_devices:
             x = torch.randn((n_iters * batch_size, n_dims_x)).float().to(device)
 
-            lin = nn.Linear(n_dims_x, n_dims_y).to(device) * 100
-            y = torch.sin(lin(x)) + torch.randn(n_iters * batch_size, n_dims_y) * 1e-4
+            lin = nn.Linear(n_dims_x, n_dims_y).to(device)
+            y = torch.sin(lin(x) * 100) + torch.randn(n_iters * batch_size, n_dims_y) * 1e-4
 
             def data_loader(i):
                 return x[i * batch_size : (i + 1) * batch_size], y[i * batch_size : (i + 1) * batch_size]