STY: Apply ruff/flake8-simplify rule SIM401

DimitriPapadopoulos · DimitriPapadopoulos · commit cb5e27011d47 · 2024-12-21T22:17:28.000+01:00
SIM401 Use `.get(..., ...)` instead of an `if` block
diff --git a/niworkflows/utils/timeseries.py b/niworkflows/utils/timeseries.py
@@ -42,7 +42,7 @@ def _cifti_timeseries(dataset):
     }
     seg = {label: [] for label in list(labels.values()) + ['Other']}
     for bm in matrix.get_index_map(1).brain_models:
-        label = 'Other' if bm.brain_structure not in labels else labels[bm.brain_structure]
+        label = labels.get(bm.brain_structure, 'Other')
         seg[label] += list(range(bm.index_offset, bm.index_offset + bm.index_count))
 
     return dataset.get_fdata(dtype='float32').T, seg

Original file line number	Diff line number	Diff line change
`@@ -42,7 +42,7 @@ def _cifti_timeseries(dataset):`
`42`	`42`	`}`
`43`	`43`	`seg = {label: [] for label in list(labels.values()) + ['Other']}`
`44`	`44`	`for bm in matrix.get_index_map(1).brain_models:`
`45`		`- label = 'Other' if bm.brain_structure not in labels else labels[bm.brain_structure]`
	`45`	`+ label = labels.get(bm.brain_structure, 'Other')`
`46`	`46`	`seg[label] += list(range(bm.index_offset, bm.index_offset + bm.index_count))`
`47`	`47`
`48`	`48`	`return dataset.get_fdata(dtype='float32').T, seg`