Update test_io.py

kornia · Mar 13, 2022 · c301d30 · c301d30
1 parent d799a3d
commit c301d30
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/test_io.py b/test/test_io.py
@@ -4,7 +4,6 @@
 from kornia_rs import Tensor as cvTensor
 
 import torch
-import jax
 import numpy as np
 
 DATA_DIR = Path(__file__).parent / "data"
@@ -38,4 +37,4 @@ def test_read_image_rs():
     # convert to dlpack to import to torch
     dlpack = K.cvtensor_to_dlpack(cv_tensor)
     th_tensor = torch.utils.dlpack.from_dlpack(dlpack)
-    assert th_tensor.shape == (195, 258, 3)
+    assert th_tensor.shape == (195, 258, 3)