Enable drop_vars for wave output

USGS-CMG · Apr 17, 2024 · dd8b066 · dd8b066
1 parent f9f53c3
commit dd8b066
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/stglib/aqd/wvsnc2waves.py b/stglib/aqd/wvsnc2waves.py
@@ -1,10 +1,7 @@
-import numpy as np
 import xarray as xr
-from tqdm import tqdm
 
-from ..core import utils, waves
+from ..core import qaqc, utils, waves
 from ..vec import nc2waves
-from . import aqdutils
 
 
 def nc_to_waves(nc_filename):
@@ -71,6 +68,8 @@ def nc_to_waves(nc_filename):
         if k in ds:
             ds = ds.drop(k)
 
+    ds = qaqc.drop_vars(ds)
+
     ds = utils.trim_max_wp(ds)
 
     ds = utils.trim_min_wh(ds)

diff --git a/stglib/rsk/nc2waves.py b/stglib/rsk/nc2waves.py
@@ -1,7 +1,7 @@
 import numpy as np
 import xarray as xr
 
-from ..core import utils, waves
+from ..core import qaqc, utils, waves
 
 
 def nc_to_waves(nc_filename):
@@ -29,6 +29,8 @@ def nc_to_waves(nc_filename):
         if k in ds:
             ds = ds.drop_vars(k)
 
+    ds = qaqc.drop_vars(ds)
+
     ds = utils.trim_max_wp(ds)
 
     ds = utils.trim_min_wh(ds)

diff --git a/stglib/vec/nc2waves.py b/stglib/vec/nc2waves.py
@@ -1,7 +1,7 @@
 import numpy as np
 import xarray as xr
 
-from ..core import utils, waves
+from ..core import qaqc, utils, waves
 
 
 def nc_to_waves(nc_filename):
@@ -59,6 +59,8 @@ def nc_to_waves(nc_filename):
         if k in ds:
             ds = ds.drop(k)
 
+    ds = qaqc.drop_vars(ds)
+
     ds = utils.trim_max_wp(ds)
 
     ds = utils.trim_min_wh(ds)