From aba2d903f187ba8e3c446cd5c847ca1e351173ec Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= Date: Thu, 20 Jun 2024 11:11:54 +0200 Subject: [PATCH] allow existing __len__ in vdataclass --- src/scgenerator/variableparameters.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/scgenerator/variableparameters.py b/src/scgenerator/variableparameters.py index 4b8b316..c3a3f16 100644 --- a/src/scgenerator/variableparameters.py +++ b/src/scgenerator/variableparameters.py @@ -439,6 +439,8 @@ else: return np.prod(self.__variables_nums__) setattr(cls, "filter", create_filter(_vfields)) - setattr(cls, "__len__", _config_len) + setattr(cls, "__vlen__", _config_len) + if not hasattr(cls, "__len__"): + setattr(cls, "__len__", _config_len) return cls