This test requires real PyTorch for tensor iteration which isn't available in CI due to module mocking. The test runs correctly in local dev with torch.
This test requires real PyTorch for tensor iteration which isn't available in CI due to module mocking. The test runs correctly in local dev with torch.