Skip to content

Commit a01e016

Browse files
Remove mps config for test (#14379)
* Remove mps config for test * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 33a5ed9 commit a01e016

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/tests_pytorch/utilities/test_fetching.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -192,9 +192,7 @@ def test_dataloader(self):
192192

193193

194194
@pytest.mark.flaky(reruns=3)
195-
@pytest.mark.parametrize(
196-
"accelerator", [pytest.param("gpu", marks=RunIf(min_cuda_gpus=1)), pytest.param("mps", marks=RunIf(mps=True))]
197-
)
195+
@pytest.mark.parametrize("accelerator", [pytest.param("cuda", marks=RunIf(min_cuda_gpus=1))])
198196
def test_trainer_num_prefetch_batches(tmpdir, accelerator):
199197

200198
model = RecommenderModel()

0 commit comments

Comments
 (0)