Added dask chunking for signal attenuation (#157)

OceanStreamIO · Dec 15, 2023 · 1aff48b · 1aff48b
1 parent 708c418
commit 1aff48b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/oceanstream/denoise/noise_masks.py b/oceanstream/denoise/noise_masks.py
@@ -41,6 +41,7 @@
             "thr": -6,
             "start": 0,
             "offset": 0,
+            "dask_chunking": {"ping_time": 100, "range_sample": 100},
         },
     },
     "impulse": {"method": "ryan", "params": {"thr": 10, "m": 5, "n": 1}},
@@ -99,6 +100,7 @@
             "thr": -5,
             "start": 0,
             "offset": 0,
+            "dask_chunking": {"ping_time": 100, "range_sample": 100},
         },
     },
     "impulse": {"method": "ryan", "params": {"thr": 3, "m": 3, "n": 1}},

diff --git a/tests/test_noise_mask.py b/tests/test_noise_mask.py
@@ -29,6 +29,7 @@
             "thr": -5,
             "start": 0,
             "offset": 0,
+            "dask_chunking": {"ping_time": 100, "range_sample": 100},
         },
     },
     "impulse": {"method": "ryan", "params": {"thr": 3, "m": 3, "n": 1}},