Skip to content

Commit d6b6c3b

Browse files
STY: Apply ruff/Perflint rule PERF401
PERF401 Use a list comprehension to create a transformed list
1 parent aeb7a8d commit d6b6c3b

File tree

4 files changed

+6
-13
lines changed

4 files changed

+6
-13
lines changed

nibabel/data.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,7 @@ def list_files(self, relative=True):
8787
for base, dirs, files in os.walk(self.base_path):
8888
if relative:
8989
base = base[len(self.base_path) + 1 :]
90-
for filename in files:
91-
out_list.append(pjoin(base, filename))
90+
out_list.extend(pjoin(base, filename) for filename in files)
9291
return out_list
9392

9493

nibabel/nicom/tests/test_dicomwrappers.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -429,10 +429,9 @@ def fake_shape_dependents(
429429

430430
class PrintBase:
431431
def __repr__(self):
432-
attr_strs = []
433-
for attr in dir(self):
434-
if attr[0].isupper():
435-
attr_strs.append(f'{attr}={getattr(self, attr)}')
432+
attr_strs = [
433+
f'{attr}={getattr(self, attr)}' for attr in dir(self) if attr[0].isupper()
434+
]
436435
return f"{self.__class__.__name__}({', '.join(attr_strs)})"
437436

438437
class DimIdxSeqElem(pydicom.Dataset):

nibabel/tests/test_euler.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,8 @@
2121
FLOAT_EPS = np.finfo(np.float64).eps
2222

2323
# Example rotations """
24-
eg_rots = []
2524
params = np.arange(-pi * 2, pi * 2.5, pi / 2)
26-
for x in params:
27-
for y in params:
28-
for z in params:
29-
eg_rots.append((x, y, z))
25+
eg_rots = [(x, y, z) for x in params for y in params for z in params]
3026

3127

3228
def x_only(x):

nibabel/tests/test_filehandles.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,7 @@ def test_multiload():
3333
tmpdir = mkdtemp()
3434
fname = pjoin(tmpdir, 'test.img')
3535
save(img, fname)
36-
for i in range(N):
37-
imgs.append(load(fname))
36+
imgs.extend(load(fname) for i in range(N))
3837
finally:
3938
del img, imgs
4039
shutil.rmtree(tmpdir)

0 commit comments

Comments
 (0)