Implement requested changes

RalfG · RalfG · commit 3961624c0087 · 2025-01-17T18:09:19.000+01:00
diff --git a/psm_utils/psm_list.py b/psm_utils/psm_list.py
@@ -100,11 +100,11 @@ def __getitem__(self, item) -> PSM | list[PSM]:
             # Return PSM property as array across full PSMList
             try:
                 return np.fromiter(
-                    [psm[item] for psm in self.psm_list], dtype=NUMPY_DTYPES[item], count=len(self)
+                    (psm[item] for psm in self.psm_list), dtype=NUMPY_DTYPES[item], count=len(self)
                 )
             except TypeError:
                 return np.fromiter(
-                    [psm[item] for psm in self.psm_list], dtype=object, count=len(self)
+                    (psm[item] for psm in self.psm_list), dtype=object, count=len(self)
                 )
         elif _is_iterable_of_bools(item):
             # Return new PSMList with items that were True

Original file line number	Diff line number	Diff line change
`@@ -100,11 +100,11 @@ def __getitem__(self, item) -> PSM \| list[PSM]:`
`100`	`100`	`# Return PSM property as array across full PSMList`
`101`	`101`	`try:`
`102`	`102`	`return np.fromiter(`
`103`		`- [psm[item] for psm in self.psm_list], dtype=NUMPY_DTYPES[item], count=len(self)`
	`103`	`+ (psm[item] for psm in self.psm_list), dtype=NUMPY_DTYPES[item], count=len(self)`
`104`	`104`	`)`
`105`	`105`	`except TypeError:`
`106`	`106`	`return np.fromiter(`
`107`		`- [psm[item] for psm in self.psm_list], dtype=object, count=len(self)`
	`107`	`+ (psm[item] for psm in self.psm_list), dtype=object, count=len(self)`
`108`	`108`	`)`
`109`	`109`	`elif _is_iterable_of_bools(item):`
`110`	`110`	`# Return new PSMList with items that were True`