Skip to content

Commit

Permalink
Revert "fix test to use the same linear layer across metric_devices"
Browse files Browse the repository at this point in the history
This reverts commit cb71355.
  • Loading branch information
kzkadc committed Sep 20, 2024
1 parent f68c8de commit b659a16
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/ignite/metrics/test_hsic.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,11 +137,10 @@ def test_integration(self, sigma_x: float, sigma_y: float):
if device.type != "xla":
metric_devices.append(device)

lin = nn.Linear(n_dims_x, n_dims_y)
for metric_device in metric_devices:
x = torch.randn((n_iters * batch_size, n_dims_x)).float().to(device)

lin.to(device)
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):
Expand Down

0 comments on commit b659a16

Please # to comment.