From 420f0ff2bdf0bdd1f29243f4371346e9c9e33bc7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= Date: Wed, 16 Aug 2023 10:56:36 +0200 Subject: [PATCH] new: support for negative prop index --- src/scgenerator/spectra.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/scgenerator/spectra.py b/src/scgenerator/spectra.py index d176594..6ecb6b8 100644 --- a/src/scgenerator/spectra.py +++ b/src/scgenerator/spectra.py @@ -157,6 +157,8 @@ class Propagation: return self._load_slice(key) if isinstance(key, (float, np.floating)): key = math.argclosest(self.parameters.compute("z_targets"), key) + elif key < 0: + key = len(self) + key array = self.io.load_spectrum(key) return Spectrum(array, self.parameters)