fix(api): add missing origin argument to noise sources

ssube · Jan 15, 2023 · 4675f89 · 4675f89
1 parent 0ed11af
commit 4675f89
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/onnx_web/image.py b/api/onnx_web/image.py
@@ -71,7 +71,7 @@ def noise_source_gaussian(source_image: Image, dims: Tuple[int, int], origin: Tu
     return noise
 
 
-def noise_source_uniform(source_image: Image, dims: Tuple[int, int]) -> Tuple[float, float, float]:
+def noise_source_uniform(source_image: Image, dims: Tuple[int, int], origin: Tuple[int, int]) -> Tuple[float, float, float]:
     width, height = dims
     size = width * height
 
@@ -93,7 +93,7 @@ def noise_source_uniform(source_image: Image, dims: Tuple[int, int]) -> Tuple[fl
     return noise
 
 
-def noise_source_normal(source_image: Image, dims: Tuple[int, int]) -> Tuple[float, float, float]:
+def noise_source_normal(source_image: Image, dims: Tuple[int, int], origin: Tuple[int, int]) -> Tuple[float, float, float]:
     width, height = dims
     size = width * height
 
@@ -115,7 +115,7 @@ def noise_source_normal(source_image: Image, dims: Tuple[int, int]) -> Tuple[flo
     return noise
 
 
-def noise_source_histogram(source_image: Image, dims: Tuple[int, int]) -> Tuple[float, float, float]:
+def noise_source_histogram(source_image: Image, dims: Tuple[int, int], origin: Tuple[int, int]) -> Tuple[float, float, float]:
     r, g, b = source_image.split()
     width, height = dims
     size = width * height