add torch minversion="1.6.0"

rapidsai · Jul 27, 2021 · 480d3c8 · 480d3c8
1 parent c31889e
commit 480d3c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/tests/test_cuda_array_interface.py b/python/cudf/cudf/tests/test_cuda_array_interface.py
@@ -170,7 +170,7 @@ def test_column_from_ephemeral_cupy_try_lose_reference():
 
 
 def test_cuda_array_interface_pytorch():
-    torch = pytest.importorskip("torch")
+    torch = pytest.importorskip("torch", minversion="1.6.0")
     if not torch.cuda.is_available():
         pytest.skip("need gpu version of pytorch to be installed")