minor change

regress_rr.py

@@ -1440,6 +1440,7 @@ def arg_parser():
                        )
     parser.add_argument('-l', '--lbl_str', type=str,
                         default='pss',
+                        choices=['pss', 'br']
                        )
     parser.add_argument('-tl', '--train_len', type=int,
                         default=5,
@@ -1518,8 +1519,7 @@ if __name__ == '__main__':
 
         for subject in subjects:
             rr_func(subject)
-
-    if subject > 0 and method == 'dsp':
+    elif subject > 0 and method == 'dsp':
         subject = subject_pre_string+str(subject).zfill(2)
         imu_rr_dsp(subject,
                    window_size=window_size,
@@ -1528,7 +1528,6 @@ if __name__ == '__main__':
                    overwrite=overwrite,
                    train_len=train_len,
                    test_standing=test_standing)
-
     elif subject <= 0 and method == 'ml':
         subjects = [subject_pre_string+str(i).zfill(2) for i in \
                     range(2, n_subject_max+1)]