Update test

Lightning-Universe · Jun 1, 2021 · 6af13d7 · 6af13d7
1 parent b691958
commit 6af13d7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/core/data/test_transforms.py b/tests/core/data/test_transforms.py
@@ -106,6 +106,7 @@ def test_kornia_parallel_transforms(with_params):
     assert transform_b.call_count == 2
 
     if with_params:
+        assert transform_a.call_args_list[1][0][1] == "test"
         # check that after the forward `_params` is set to None
         assert transform_a._params == transform_a._params is None