Transform tests: min Kornia version, filter warnings (#306)

microsoft · Dec 19, 2021 · 6ffb9b9 · 6ffb9b9
1 parent 13804e0
commit 6ffb9b9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -73,6 +73,7 @@ filterwarnings = [
     "ignore:.*Create unlinked descriptors is going to go away.*:DeprecationWarning",
     # https://github.com/tensorflow/tensorboard/pull/5138
     "ignore:.*is a deprecated alias for the builtin.*:DeprecationWarning",
+    "ignore:.*Previous behaviour produces incorrect box coordinates.*:UserWarning",
 ]
 markers = [
     "slow: marks tests as slow",

diff --git a/tests/transforms/test_transforms.py b/tests/transforms/test_transforms.py
@@ -11,6 +11,9 @@
 
 from torchgeo.transforms import indices, transforms
 
+# Tests require newer version of Kornia for newer bounding box behavior
+pytest.importorskip("kornia", minversion="0.6.2")
+
 
 @pytest.fixture
 def sample() -> Dict[str, Tensor]: