Fix ruff in uts

openvinotoolkit · Aug 12, 2024 · 97558ad · 97558ad
1 parent 10a557b
commit 97558ad
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tests/unit/core/model/test_visual_prompting.py b/tests/unit/core/model/test_visual_prompting.py
@@ -536,7 +536,9 @@ def test_customize_outputs_training(
             [
                 {
                     1: PredictedMask(mask=[[1, 2, 3], [4, 5, 6]], points=[[13, 14, 15], [16, 17, 18]], scores=[1, 1]),
-                    2: PredictedMask(mask=[[7, 8, 9], [10, 11, 12]], points=[[19, 20, 21], [22, 23, 24]], scores=[1, 1]),
+                    2: PredictedMask(
+                        mask=[[7, 8, 9], [10, 11, 12]], points=[[19, 20, 21], [22, 23, 24]], scores=[1, 1]
+                    ),
                 },
             ],
             [
@@ -547,11 +549,15 @@ def test_customize_outputs_training(
             [
                 {
                     1: PredictedMask(mask=[[1, 2, 3], [4, 5, 6]], points=[[13, 14, 15], [16, 17, 18]], scores=[1, 1]),
-                    2: PredictedMask(mask=[[7, 8, 9], [10, 11, 12]], points=[[19, 20, 21], [22, 23, 24]], scores=[1, 1]),
+                    2: PredictedMask(
+                        mask=[[7, 8, 9], [10, 11, 12]], points=[[19, 20, 21], [22, 23, 24]], scores=[1, 1]
+                    ),
                 },
                 {
                     1: PredictedMask(mask=[[1, 2, 3], [4, 5, 6]], points=[[13, 14, 15], [16, 17, 18]], scores=[1, 1]),
-                    2: PredictedMask(mask=[[7, 8, 9], [10, 11, 12]], points=[[19, 20, 21], [22, 23, 24]], scores=[1, 1]),
+                    2: PredictedMask(
+                        mask=[[7, 8, 9], [10, 11, 12]], points=[[19, 20, 21], [22, 23, 24]], scores=[1, 1]
+                    ),
                 },
             ],
         ],