File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -404,7 +404,7 @@ def from_cnv(fname):
404
404
f .close ()
405
405
406
406
prkeys = [
407
- "prM " ,
407
+ "prM" ,
408
408
"prE" ,
409
409
"prDM" ,
410
410
"pr50M" ,
@@ -415,9 +415,14 @@ def from_cnv(fname):
415
415
"depSM" ,
416
416
"prDE" ,
417
417
]
418
+ df .columns = df .columns .str .strip ()
418
419
prkey = [key for key in prkeys if key in df .columns ]
419
- if len (prkey ) != 1 :
420
- raise ValueError (f"Expected one pressure/depth column, got { prkey } ." )
420
+ if len (prkey ) == 0 :
421
+ raise ValueError ("Expected one pressure/depth column, didn't receive any" )
422
+ elif len (prkey ) > 1 :
423
+ # if multiple keys present then keep the first one
424
+ prkey = prkey [0 ]
425
+
421
426
df .set_index (prkey , drop = True , inplace = True )
422
427
df .index .name = "Pressure [dbar]"
423
428
if prkey == "depSM" :
You can’t perform that action at this time.
0 commit comments