From 5bc8802ca9d8f2bc5f30995970d8c7918ec23557 Mon Sep 17 00:00:00 2001 From: Sam-Armstrong Date: Sun, 30 Jun 2024 03:53:51 +0100 Subject: [PATCH] add failing test --- ivy_tests/test_integrations/test_kornia.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/ivy_tests/test_integrations/test_kornia.py b/ivy_tests/test_integrations/test_kornia.py index 5650bd765bff..8a90be0f8015 100644 --- a/ivy_tests/test_integrations/test_kornia.py +++ b/ivy_tests/test_integrations/test_kornia.py @@ -8,19 +8,13 @@ # TODO: remove -def test_sharpness(target_framework, mode, backend_compile): - trace_args = ( - torch.rand(1, 1, 5, 5), - 0.5, - ) +def test_equalize(target_framework, mode, backend_compile): + trace_args = (torch.rand(1, 2, 3, 3),) trace_kwargs = {} - test_args = ( - torch.rand(5, 1, 5, 5), - 1.0, - ) + test_args = (torch.rand(5, 2, 3, 3),) test_kwargs = {} _test_function( - kornia.enhance.sharpness, + kornia.enhance.equalize, trace_args, trace_kwargs, test_args,