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