MAINT: allow old clean routine if needed

pysatNASA/instruments/cnofs_vefi.py

@@ -62,6 +62,7 @@
 
 import datetime as dt
 import functools
+import numpy as np
 
 from pysat.instruments.methods import general as mm_gen
 
@@ -90,8 +91,24 @@
 init = functools.partial(mm_nasa.init, module=mm_cnofs, name=name)
 
 
-# Use default clean
-clean = mm_nasa.clean
+def clean(self):
+    """Clean VEFI data to the specified level.
+
+    Note
+    ----
+    'dusty' or 'clean' removes data when interpolation flag is set to 1
+    'dirty' is the same as 'none'
+
+    """
+
+    if 'B_flag' in self.variables:
+        if (self.clean_level == 'dusty') | (self.clean_level == 'clean'):
+            idx, = np.where(self['B_flag'] == 0)
+            self.data = self[idx, :]
+    else:
+        mm_nasa.clean(self)
+
+    return
 
 
 # ----------------------------------------------------------------------------