From 96ba897873028e721042f97cf9f1e8a12c94888f Mon Sep 17 00:00:00 2001 From: BanzaiTokyo Date: Mon, 14 Apr 2025 13:40:59 +0000 Subject: [PATCH] adds available_device to RootMeanSquaredError #3335 --- tests/ignite/metrics/test_root_mean_squared_error.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/ignite/metrics/test_root_mean_squared_error.py b/tests/ignite/metrics/test_root_mean_squared_error.py index ebdd84aa08d4..e5dcb25cca07 100644 --- a/tests/ignite/metrics/test_root_mean_squared_error.py +++ b/tests/ignite/metrics/test_root_mean_squared_error.py @@ -29,8 +29,9 @@ def test_data(request): @pytest.mark.parametrize("n_times", range(3)) -def test_compute(n_times, test_data): - rmse = RootMeanSquaredError() +def test_compute(n_times, test_data, available_device): + rmse = RootMeanSquaredError(device=available_device) + assert rmse._device == torch.device(available_device) y_pred, y, batch_size = test_data rmse.reset()