From de99c46bd8a8c8bb8788001464f0d58aee442d59 Mon Sep 17 00:00:00 2001
From: Pritwiraj Moulik <pritwiraj.moulik@gmail.com>
Date: Sun, 3 Dec 2023 23:06:23 -0500
Subject: [PATCH] minor fixes to mode catalogs

---
 avni/data/NM.py | 9 +++++----
 avni/data/SW.py | 3 ++-
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/avni/data/NM.py b/avni/data/NM.py
index 3c92280..496fe6b 100644
--- a/avni/data/NM.py
+++ b/avni/data/NM.py
@@ -16,7 +16,7 @@
 #import matplotlib.pyplot as plt
 from avni.models import Reference1D
 from avni import constants,tools
-import pint
+import pint_pandas
 import warnings
 
 if sys.version_info[0] >= 3: unicode = str
@@ -40,7 +40,7 @@ def read_rts_catalog(infile, base_units = True):
     ilev_flag = []
 
     # reference1D instance and the units class
-    pint.PintType.ureg = constants.ureg
+    pint_pandas.PintType.ureg = constants.ureg
     ref1d = Reference1D()
 
     #open buffer
@@ -95,7 +95,7 @@ def read_rts_catalog(infile, base_units = True):
 
     # loop over names and call evaluate_at_depth
     # Create data array for converted to Panda array with units
-    PA_ = pint.PintArray; temp_dict = {}
+    PA_ = pint_pandas.PintArray; temp_dict = {}
     for paraindx,param in enumerate(names):
         print(paraindx,param)
 
@@ -111,7 +111,8 @@ def read_rts_catalog(infile, base_units = True):
     modelarr = pd.DataFrame(temp_dict)
     if base_units: # convert to base units
         for col in modelarr.columns: modelarr[col] = modelarr[col].pint.to_base_units()
-    modelarr['depth'] = PA_((constants.R.magnitude - modelarr['radius'].pint.to(constants.R.units).data).tolist(), dtype = constants.R.units)
+
+    modelarr['depth'] = PA_(constants.R.magnitude - modelarr['radius'].pint.to(constants.R.units).values.quantity.magnitude, dtype = constants.R.units)
 
     # number of layers
     nlev_out=struct.unpack(ifswp+'i',f.read(4))[0]; cc += 4
diff --git a/avni/data/SW.py b/avni/data/SW.py
index a2bc04b..88b1f60 100644
--- a/avni/data/SW.py
+++ b/avni/data/SW.py
@@ -14,6 +14,7 @@
 import h5py
 import time
 import warnings
+import pdb
 from .. import constants
 
 if sys.version_info[0] >= 3: unicode = str
@@ -102,7 +103,7 @@ def get_velocity(table,period,overtone,mode_type,output='pvel'):
     vel = table[mode_type][str(overtone)].attrs[output]
 
     if omega_query < np.min(omega) or omega_query > np.max(omega):
-        raise ValueError('period {} doesnt exist in table for requested mode'.format(period))
+        warnings.warn('period {} doesnt exist in table bounds for requested mode [{},{}]. Extroplating...'.format(period,(2.*np.pi)/np.max(omega),(2.*np.pi)/np.min(omega)))
 
     # Perform interpolation to get velocity for requested period
     vel_i = np.interp(omega_query,omega,vel)