Add issue link

microsoft · Jun 30, 2024 · b794e34 · b794e34
1 parent bec9bab
commit b794e34
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/transforms/test_color.py b/tests/transforms/test_color.py
@@ -37,6 +37,7 @@ def test_random_grayscale_sample(weights: Tensor, sample: dict[str, Tensor]) ->
     aug = K.AugmentationSequential(
         RandomGrayscale(weights, p=1), keepdim=True, data_keys=None
     )
+    # https://github.com/kornia/kornia/issues/2848
     aug.keepdim = True
     output = aug(sample)
     assert output['image'].shape == sample['image'].shape