From 7822d4ac9e438ea1097b8694f200a7f7aa9c1790 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micka=C3=ABl=20Bonnin?= <mickael.bonnin@univ-nantes.fr> Date: Thu, 15 Sep 2022 09:28:00 +0200 Subject: [PATCH] Remove 'data = ' in call of 'spectro_extract' --- run.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/run.py b/run.py index d9692f8..ef9fa5b 100644 --- a/run.py +++ b/run.py @@ -48,16 +48,16 @@ def read_args() -> argparse.Namespace: def main(args: argparse.Namespace): if args.mode == 'pred': - data = spectro_extract(data_dir=args.data_dir, - spectro_dir=args.spectro_dir) + spectro_extract(data_dir=args.data_dir, + spectro_dir=args.spectro_dir) pred(model_dir=args.model_dir, spectro_dir=args.spectro_dir, output_dir=args.output_dir) if args.mode == 'valid': events = np.genfromtxt( f'{args.csv_dir}', delimiter=',', skip_header=1, dtype=str) - data = spectro_extract(data_dir=args.data_dir, - spectro_dir=args.spectro_dir, events_list=events) + spectro_extract(data_dir=args.data_dir, + spectro_dir=args.spectro_dir, events_list=events) valid(model_dir=args.model_dir, spectro_dir=args.spectro_dir, output_dir=args.output_dir, event_label=events) -- GitLab