Skip to content

Commit

Permalink
Reformatted the file.
Browse files Browse the repository at this point in the history
  • Loading branch information
Benteng Ma committed Jun 17, 2024
1 parent a405f20 commit 46170e4
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -600,7 +600,9 @@ def binary_erosion_dilation(

# Check if the length of thresholds matches the number of channels
if len(thresholds) != tensor.size(1):
raise ValueError(f"Length of thresholds {len(thresholds)} must match the number of channels {tensor.size(1)}")
raise ValueError(
f"Length of thresholds {len(thresholds)} must match the number of channels {tensor.size(1)}"
)

# Binary thresholding
for i, threshold in enumerate(thresholds):
Expand Down

0 comments on commit 46170e4

Please sign in to comment.