From d1d46b721f100cbf1db99eae276adaaf0f78858a Mon Sep 17 00:00:00 2001 From: BaptisteVandecrux Date: Thu, 13 Jun 2024 10:28:47 +0200 Subject: [PATCH] fixed function names --- src/pypromice/process/get_l2tol3.py | 4 ++-- src/pypromice/process/join_l2.py | 5 +++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/pypromice/process/get_l2tol3.py b/src/pypromice/process/get_l2tol3.py index a431c44b..e8ed5a96 100644 --- a/src/pypromice/process/get_l2tol3.py +++ b/src/pypromice/process/get_l2tol3.py @@ -40,8 +40,8 @@ def get_l2tol3(): m = getMeta(args.metadata) # Define Level 2 dataset from file - l2 = xr.open_dataset(args.inpath) - + with xr.open_dataset(args.inpath) as l2: + l2.load() if 'bedrock' in l2.attrs.keys(): l2.attrs['bedrock'] = l2.attrs['bedrock'] == 'True' if 'number_of_booms' in l2.attrs.keys(): diff --git a/src/pypromice/process/join_l2.py b/src/pypromice/process/join_l2.py index 71259c18..28bf2e0e 100644 --- a/src/pypromice/process/join_l2.py +++ b/src/pypromice/process/join_l2.py @@ -28,7 +28,8 @@ def loadArr(infile): df = pd.read_csv(infile, index_col=0, parse_dates=True) ds = xr.Dataset.from_dataframe(df) elif infile.split('.')[-1].lower() == 'nc': - ds = xr.open_dataset(infile) + with xr.open_dataset(infile) as ds: + ds.load() try: name = ds.attrs['station_id'] @@ -44,7 +45,7 @@ def loadArr(infile): return ds, name -def join_levels(): +def join_l2(): args = parse_arguments_join() # Define variables and metadata (either from file or pypromice package defaults)