Skip to content

Commit

Permalink
Refactor code using linter
Browse files Browse the repository at this point in the history
  • Loading branch information
proneetsharma committed Jul 6, 2021
1 parent c26d013 commit d63f2af
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
1 change: 0 additions & 1 deletion paz/processors/draw.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,4 +133,3 @@ def __init__(self, max_radius_scale=.5):

def call(self, image):
return draw_random_polygon(image)

4 changes: 3 additions & 1 deletion tests/paz/processors/draw_test.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,14 @@
import pytest
from paz import processors as pr


def test_DrawBoxes2D_with_invalid_class_names_type():
with pytest.raises(TypeError):
class_names = 'Face'
colors = [[255, 0, 0]]
pr.DrawBoxes2D(class_names, colors)



def test_DrawBoxes2D_with_invalid_colors_type():
with pytest.raises(TypeError):
class_names = ['Face']
Expand Down

0 comments on commit d63f2af

Please sign in to comment.