Skip to content

Commit

Permalink
Fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
ilyag-grovety committed Apr 4, 2023
1 parent c9e6ed1 commit 624b5d2
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 2 deletions.
5 changes: 4 additions & 1 deletion tests/python/contrib/test_ethosu/test_codegen.py
Original file line number Diff line number Diff line change
Expand Up @@ -1101,7 +1101,10 @@ def test_tflite_resize2d_nearest_neighbor(accel_type, ifm_shape, size, half_pixe
@tf.function
def resize_model(x):
return tf.compat.v1.image.resize_nearest_neighbor(
x, size, align_corners=align_corners, half_pixel_centers=half_pixel,
x,
size,
align_corners=align_corners,
half_pixel_centers=half_pixel,
)

infra.compare_tvm_with_tflite(
Expand Down
5 changes: 4 additions & 1 deletion tests/python/contrib/test_ethosu/test_legalize.py
Original file line number Diff line number Diff line change
Expand Up @@ -2359,7 +2359,10 @@ def create_tflite_graph():
@tf.function
def resize_model(x):
return tf.compat.v1.image.resize_nearest_neighbor(
x, size, align_corners=align_corners, half_pixel_centers=half_pixel,
x,
size,
align_corners=align_corners,
half_pixel_centers=half_pixel,
)

concrete_func = resize_model.get_concrete_function(
Expand Down

0 comments on commit 624b5d2

Please sign in to comment.