fix flake

tests/unittests/image/test_tv.py

@@ -72,7 +72,7 @@ class TestTotalVariation(MetricTester):
     @pytest.mark.parametrize("dist_sync_on_step", [True, False])
     def test_total_variation(self, preds, target, reduction, ddp, dist_sync_on_step):
         """Test modular implementation."""
-        if reduction == None and ddp:
+        if reduction is None and ddp:
             pytest.skip("reduction=None and ddp=True runs out of memory on CI hardware, but it does work")
         self.run_class_metric_test(
             ddp,