diff --git a/pyproject.toml b/pyproject.toml index a63cac86..79dafe73 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -150,7 +150,7 @@ lint.per-file-ignores."docs/*" = [ "I" ] lint.per-file-ignores."tests/*" = [ "D" ] lint.pydocstyle.convention = "numpy" -[tool.pytest.ini_option] +[tool.pytest.ini_options] markers = [ "gpu: mark test to run on GPU" ] testpaths = [ "tests" ] xfail_strict = true diff --git a/tests/test_dataset.py b/tests/test_dataset.py index 33892ae2..299c7d5d 100644 --- a/tests/test_dataset.py +++ b/tests/test_dataset.py @@ -413,7 +413,7 @@ def test_3d( import torch assert isinstance(x, torch.Tensor) - x = np.array(x.cpu()) + x = x.cpu().numpy() x_list.append(x) idx_list.append(idxs.ravel()) x = np.vstack(x_list)