Use 'spectro_extract' instead of 'valid' & 'pred'.

run.py

@@ -6,7 +6,7 @@ import argparse
 import numpy as np
 
 from prediction import pred, valid
-from data_process import spectro_extract_pred, spectro_extract_valid
+from data_process import spectro_extract
 
 
 def read_args() -> argparse.Namespace:
@@ -48,16 +48,16 @@ def read_args() -> argparse.Namespace:
 def main(args: argparse.Namespace):
 
     if args.mode == 'pred':
-        data = spectro_extract_pred(
-            data_dir=args.data_dir, spectro_dir=args.spectro_dir)
+        data = 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)
 
     elif args.mode == 'valid':
         events = np.genfromtxt(
             f'{args.csv_dir}', delimiter=',', skip_header=1, dtype=str)
-        data = spectro_extract_valid(
-            data_dir=args.data_dir, spectro_dir=args.spectro_dir, events_list=events)
+        data = spectro_extract(data_dir=args.data_dir,
+                               spectro_dir=args.spectro_dir, events_list=events)
         valid(model_dir=args.model_dir, spectro_dir='./spectro_demo',
               output_dir=args.output_dir, event_label=events)