Simplify _get_item_metadata

list comprehension is little clunky but avoids the 1-length special case: cleaner overall
pace-neutrons · Sep 20, 2024 · 87f32a0 · 87f32a0
1 parent aaaa25a
commit 87f32a0
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/euphonic/spectra.py b/euphonic/spectra.py
@@ -865,10 +865,8 @@ def _get_item_metadata(self, item):  # noqa: F811
             return metadata_lines[item]
         if isinstance(item, slice):
             return self._combine_metadata(metadata_lines[item])
-        if len(item) == 1:
-            return metadata_lines[item[0]]
-        return self._combine_metadata(
-            list(itemgetter(*item)(metadata_lines)))
+        # Item must be some kind of integer sequence
+        return self._combine_metadata([metadata_lines[i] for i in item])
 
     def copy(self) -> Self:
         """Get an independent copy of spectrum"""