Skip to content

Commit 7122f27

Browse files
authored
Added available_device in metric tests (#3335) (#3354)
* Added available_device in metric tests (#3335) * Added available_device fixture in test_mean_absolute_error * Added available_device fixture in test_mean_mean_pairwise_distance * Apply formatting to previous changes
1 parent c48d9f9 commit 7122f27

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tests/ignite/metrics/test_mean_absolute_error.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,9 @@ def test_case(request):
2929

3030

3131
@pytest.mark.parametrize("n_times", range(5))
32-
def test_compute(n_times, test_case):
33-
mae = MeanAbsoluteError()
32+
def test_compute(n_times, test_case, available_device):
33+
mae = MeanAbsoluteError(device=available_device)
34+
assert mae._device == torch.device(available_device)
3435

3536
y_pred, y, batch_size = test_case
3637

tests/ignite/metrics/test_mean_pairwise_distance.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,9 @@ def test_case(request):
2929

3030

3131
@pytest.mark.parametrize("n_times", range(5))
32-
def test_compute(n_times, test_case):
33-
mpd = MeanPairwiseDistance()
32+
def test_compute(n_times, test_case, available_device):
33+
mpd = MeanPairwiseDistance(device=available_device)
34+
assert mpd._device == torch.device(available_device)
3435

3536
y_pred, y, batch_size = test_case
3637

0 commit comments

Comments
 (0)