Skip to content

Commit 678405d

Browse files
authored
Added available_device fixture in test_mean_squared_error (#3335) (#3341)
1 parent 3c75ce0 commit 678405d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/ignite/metrics/test_mean_squared_error.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ def test_case(request):
2929

3030

3131
@pytest.mark.parametrize("n_times", range(5))
32-
def test_compute(n_times, test_case):
33-
mse = MeanSquaredError()
32+
def test_compute(n_times, available_device, test_case):
33+
mse = MeanSquaredError(available_device)
3434

3535
y_pred, y, batch_size = test_case
3636

0 commit comments

Comments
 (0)