Fix for ensemble table

LinkedEarth · Oct 17, 2023 · 4870bb2 · 4870bb2
1 parent 8980454
commit 4870bb2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/pylipd/lipd.py b/pylipd/lipd.py
@@ -881,9 +881,11 @@ def get_ensemble_tables(self, dsname = None, ensembleVarName = None, ensembleDep
             query = query.replace("[ensembleDepthVarName]", ensembleDepthVarName)
 
         qres, qres_df = self.query(query)
-
+
+        nan_replace = re.compile(re.escape('NaN'), re.IGNORECASE)
+
         qres_df['ensembleDepthValues']=qres_df['ensembleDepthValues'].apply(lambda row : np.fromstring(row.strip("[]"), sep=','))
-        qres_df['ensembleVariableValues']=qres_df['ensembleVariableValues'].apply(lambda row : np.array(ast.literal_eval(row)))
+        qres_df['ensembleVariableValues']=qres_df['ensembleVariableValues'].apply(lambda row : np.array(ast.literal_eval(nan_replace.sub(row, 'None'))))        
 
 
         return qres_df

diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = pylipd
-version = 1.3.7.b0
+version = 1.3.7.b1
 author = Varun Ratnakar, Deborah Khider
 author_email = [email protected]
 description= Python utilities for handling LiPD data

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 import os
 from setuptools import setup, find_packages
 
-version = '1.3.7.b0'
+version = '1.3.7.b1'
 
 # Read the readme file contents into variable
 def read(fname):