ENH: Rename the csv module to prevent errors

ARM-DOE · Jan 16, 2024 · 7409cce · 7409cce
1 parent 6eda11b
commit 7409cce
Show file tree

Hide file tree

Showing 6 changed files with 678 additions and 352 deletions.
diff --git a/act/io/__init__.py b/act/io/__init__.py
@@ -7,7 +7,6 @@
 
 __getattr__, __dir__, __all__ = lazy.attach(
     __name__,
-
     submodules=['arm', 'csv', 'icartt', 'mpl', 'neon', 'noaagml', 'noaapsl', 'pysp2'],
     submod_attrs={
         'arm': [
@@ -18,7 +17,7 @@
             'check_if_tar_gz_file',
             'read_arm_mmcr',
         ],
-        'csv': ['read_csv'],
+        'text': ['read_csv'],
         'icartt': ['read_icartt'],
         'mpl': ['proc_sigma_mplv5_read', 'read_sigma_mplv5'],
         'neon': ['read_neon_csv'],
@@ -38,9 +37,7 @@
             'read_psl_radar_fmcw_moment',
             'read_psl_surface_met',
         ],
-        'pysp2': ['read_hk_file', 'read_sp2', 'read_sp2_dat'
-        ],
-        'sodar' : [
-            'read_mfas_sodar']
+        'pysp2': ['read_hk_file', 'read_sp2', 'read_sp2_dat'],
+        'sodar': ['read_mfas_sodar'],
     },
 )
diff --git a/act/io/csv.py b/act/io/csv.py
diff --git a/act/io/neon.py b/act/io/neon.py
@@ -2,11 +2,13 @@
 Modules for reading in NOAA PSL data.
 """
 
-import pandas as pd
-import xarray as xr
 import datetime as dt
+
 import numpy as np
-from act.io.csv import read_csv
+import pandas as pd
+import xarray as xr
+
+from act.io.text import read_csv
 
 
 def read_neon_csv(files, variable_files=None, position_files=None):
@@ -85,8 +87,18 @@ def read_neon_csv(files, variable_files=None, position_files=None):
                 ds['lat'] = xr.DataArray(data=float(loc_df['referenceLatitude'].values[idx]))
                 ds['lon'] = xr.DataArray(data=float(loc_df['referenceLongitude'].values[idx]))
                 ds['alt'] = xr.DataArray(data=float(loc_df['referenceElevation'].values[idx]))
-                variables = ['xOffset', 'yOffset', 'zOffset', 'eastOffset', 'northOffset',
-                             'pitch', 'roll', 'azimuth', 'xAzimuth', 'yAzimuth']
+                variables = [
+                    'xOffset',
+                    'yOffset',
+                    'zOffset',
+                    'eastOffset',
+                    'northOffset',
+                    'pitch',
+                    'roll',
+                    'azimuth',
+                    'xAzimuth',
+                    'yAzimuth',
+                ]
                 for v in variables:
                     ds[v] = xr.DataArray(data=float(loc_df[v].values[idx]))
         multi_ds.append(ds)