Skip to content

Commit 1d7c355

Browse files
Remove code to save image for test
1 parent 9f9e320 commit 1d7c355

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

test/test_utils.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -137,11 +137,9 @@ def test_draw_boxes_with_coloured_label_backgrounds():
137137
labels = ["a", "b", "c", "d"]
138138
colors = ["green", "#FF00FF", (0, 255, 0), "red"]
139139
label_colors = ["green", "red", (0, 255, 0), "#FF00FF"]
140-
result = utils.draw_bounding_boxes(img, boxes, labels=labels, colors=colors, fill=True, label_colors=label_colors, fill_labels=True)
141-
# utils.save_image(
142-
# result.div(255),
143-
# "/home/antoinesimoulin/vision/test/assets/fakedata/draw_boxes_different_label_fill_colors.png",
144-
# )
140+
result = utils.draw_bounding_boxes(
141+
img, boxes, labels=labels, colors=colors, fill=True, label_colors=label_colors, fill_labels=True
142+
)
145143

146144
path = os.path.join(
147145
os.path.dirname(os.path.abspath(__file__)), "assets", "fakedata", "draw_boxes_different_label_fill_colors.png"

0 commit comments

Comments
 (0)