Skip to content

Commit

Permalink
fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
MateoLostanlen committed Apr 9, 2024
1 parent 92667b1 commit 2b631bf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_vision.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ def test_classifier(mock_wildfire_image):
out, pad = model.preprocess_image(mock_wildfire_image)
assert isinstance(out, np.ndarray) and out.dtype == np.float32
assert out.shape == (1, 3, 384, 640)
assert isinstance(pad, tuple)
assert isinstance(pad, tuple)
# Check inference
out = model(mock_wildfire_image)
assert out.shape == (1, 5)
Expand Down

0 comments on commit 2b631bf

Please sign in to comment.