Skip to content

Commit 6e00396

Browse files
committed
format with black
1 parent 5a4e1a6 commit 6e00396

File tree

1 file changed

+22
-4
lines changed

1 file changed

+22
-4
lines changed

petdeface/petdeface.py

Lines changed: 22 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -341,10 +341,14 @@ def init_single_subject_wf(
341341
if subject_data is None:
342342
raise FileNotFoundError(f"Could not find data for subject sub-{subject_id}")
343343

344-
# we combine the sessions to include and exclude into a single set of sessions to exclude from
344+
# we combine the sessions to include and exclude into a single set of sessions to exclude from
345345
# the set of all sessions
346346
if session:
347-
sessions_to_exclude = list(set(bids_data.get_sessions()) - (set(bids_data.get_sessions()) & set(session)) | set(exclude_session))
347+
sessions_to_exclude = list(
348+
set(bids_data.get_sessions())
349+
- (set(bids_data.get_sessions()) & set(session))
350+
| set(exclude_session)
351+
)
348352
else:
349353
sessions_to_exclude = exclude_session
350354

@@ -918,8 +922,22 @@ def cli():
918922
required=False,
919923
default=[],
920924
)
921-
parser.add_argument("--session", help="Select only a specific session(s) to include in the defacing workflow", type=str, nargs="+", required=False, default=[])
922-
parser.add_argument("--exclude_session", help="Select a specific session(s) to exclude from the defacing workflow", type=str, nargs="+", required=False, default=[])
925+
parser.add_argument(
926+
"--session",
927+
help="Select only a specific session(s) to include in the defacing workflow",
928+
type=str,
929+
nargs="+",
930+
required=False,
931+
default=[],
932+
)
933+
parser.add_argument(
934+
"--exclude_session",
935+
help="Select a specific session(s) to exclude from the defacing workflow",
936+
type=str,
937+
nargs="+",
938+
required=False,
939+
default=[],
940+
)
923941

924942
arguments = parser.parse_args()
925943
return arguments

0 commit comments

Comments
 (0)