FIX: overwrite

scil-vital · Apr 4, 2024 · 795ed57 · 795ed57
1 parent a636ce3
commit 795ed57
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TractOracleNet/runners/predictor.py b/TractOracleNet/runners/predictor.py
@@ -10,7 +10,7 @@
 from tqdm import tqdm
 
 from scilpy.io.utils import (
-    assert_inputs_exist, assert_outputs_exist)
+    assert_inputs_exist, assert_outputs_exist, add_overwrite_arg)
 
 from TractOracleNet.utils import get_data, save_filtered_streamlines
 from TractOracleNet.models.utils import get_model
@@ -199,6 +199,7 @@ def _build_arg_parser(parser):
                         ' Streamlines\' endpoints should be uniformized for'
                         ' best visualization.')
 
+    add_overwrite_arg(parser)
 
 def parse_args():
     """ Filter a tractogram. """