fix test_kornia_parallel_transforms

Lightning-Universe · Jun 1, 2021 · b691958 · b691958
1 parent 8f71343
commit b691958
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/data/test_transforms.py b/tests/core/data/test_transforms.py
@@ -107,7 +107,7 @@ def test_kornia_parallel_transforms(with_params):
 
     if with_params:
         # check that after the forward `_params` is set to None
-        assert transform_a._params == transform_a._params == None
+        assert transform_a._params == transform_a._params is None
 
     assert torch.allclose(transform_a.call_args_list[0][0][0], samples[0])
     assert torch.allclose(transform_a.call_args_list[1][0][0], samples[1])