add return values, remove embed

This commit is contained in:
Jan Grewe 2018-11-12 11:32:50 +01:00
parent 08402fcf13
commit 5b6f2bd967

View File

@ -5,8 +5,6 @@ from IPython import embed
def read_baseline_eod(dataset):
base = dataset.split(os.path.sep)[-1] + ".nix"
nix_file = nix.File.open(os.path.join(dataset, base))
embed()
nix_file.close()
b = nix_file.blocks[0]
for t in b.tags():
if "baseline" in t.name:
@ -15,11 +13,11 @@ def read_baseline_eod(dataset):
eod = t.retrieve_data("LocalEOD-1")
time = eod_da.dimensions[0].axis(len(eod))
nix_file.close()
return time, eod
if __name__ == "__main__":
data_dir = "../data"
dataset = "2018-11-09-ad-invivo-1"
eod = read_baseline_eod(os.path.join(data_dir, dataset))
time, eod = read_baseline_eod(os.path.join(data_dir, dataset))