Skip to content

Commit

Permalink
Fix noise level range check
Browse files Browse the repository at this point in the history
Signed-off-by: ArchieMeng <[email protected]>
  • Loading branch information
ArchieMeng committed Mar 9, 2022
1 parent ffd038d commit 9f80def
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion waifu2x_ncnn_vulkan_python/waifu2x_ncnn_vulkan.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ def __init__(

# check arguments' validity
assert scale in (1, 2), "scale must be 1 or 2"
assert noise in range(4), "noise must be 0-4"
assert noise in range(-1, 4), "noise must be [-1, 3]"
assert isinstance(tta_mode, bool), "tta_mode isn't a boolean value"
assert num_threads >= 1, "num_threads must be an integer >= 1"

Expand Down

0 comments on commit 9f80def

Please sign in to comment.