Fix exclude argument to write_statsfile in CerebNet/inference.py

Deep-MI · Aug 21, 2024 · ffe2286 · ffe2286
1 parent a1cd120
commit ffe2286
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/CerebNet/inference.py b/CerebNet/inference.py
@@ -502,7 +502,7 @@ def run(self, subject_dirs: SubjectList):
                                 normfile=norm_file,
                                 lut=self.freesurfer_lut_file,
                                 volume_precision="3",
-                                exclude=0,
+                                exclude=[0],
                                 pvfile=norm_file,
                                 report_empty=True,
                                 extra_header=[],

diff --git a/FastSurferCNN/segstats.py b/FastSurferCNN/segstats.py
@@ -1324,6 +1324,9 @@ def _table_body(file: IO, _dataframe: pd.DataFrame) -> None:
     if segfile is not None and not isinstance(segfile, Path):
         segfile = Path(segfile)
 
+    if exclude is not None and not isinstance(exclude, Sequence):
+        raise RuntimeError("exclude must be a sequence of ints or None!")
+
     segstatsfile.parent.mkdir(exist_ok=True)
     with open(segstatsfile, "w") as fp:
         _title(fp)