From 90717ba85d6d9715dd4aae016e17da6995e08906 Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Thu, 27 Jan 2022 17:01:54 +0000 Subject: [PATCH] fix: 1D-Corr objects can now be initialized with None entries. This fixes a bug in Corr.m_eff with padding. --- pyerrors/correlators.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pyerrors/correlators.py b/pyerrors/correlators.py index 2209c431..f051855a 100644 --- a/pyerrors/correlators.py +++ b/pyerrors/correlators.py @@ -42,9 +42,9 @@ class Corr: if not isinstance(data_input, list): raise TypeError('Corr__init__ expects a list of timeslices.') - if all([(isinstance(item, Obs) or isinstance(item, CObs)) for item in data_input]): - _assert_equal_properties(data_input) - self.content = [np.asarray([item]) for item in data_input] + if all([(isinstance(item, Obs) or isinstance(item, CObs)) or item is None for item in data_input]): + _assert_equal_properties([o for o in data_input if o is not None]) + self.content = [np.asarray([item]) if item is not None else None for item in data_input] self.N = 1 elif all([isinstance(item, np.ndarray) or item is None for item in data_input]) and any([isinstance(item, np.ndarray) for item in data_input]):