Merge pull request #106 from hassonlab/hvgazula/issue97

fix #97
hassonlab · Dec 12, 2022 · 97c2717 · 97c2717
2 parents b0319f0 + 3e22d9d
commit 97c2717
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/scripts/tfspkl_build_matrices.py b/scripts/tfspkl_build_matrices.py
@@ -53,10 +53,8 @@ def build_design_matrices(CONFIG, delimiter=","):
             df = pd.DataFrame(sigelec_list, columns=["subject", "electrode"])
         except:
             # If the electrode file is in the new format
-            df = pd.read_csv(
-                CONFIG["sig_elec_file"], columns=["subject", "electrode"]
-            )
-        else:
+            df = pd.read_csv(CONFIG["sig_elec_file"])
+        finally:
             electrodes_dict = (
                 df.groupby("subject")["electrode"].apply(list).to_dict()
             )

diff --git a/scripts/tfspkl_parser.py b/scripts/tfspkl_parser.py
@@ -42,4 +42,7 @@ def arg_parser():
 
     args = parser.parse_args()
 
+    if not args.subject:
+        args.subject = 777
+
     return args