From d61c15e825c80a3088d03aeac2d66c4fa32e3275 Mon Sep 17 00:00:00 2001 From: Martin Glatzle Date: Fri, 22 Feb 2019 17:07:42 +0100 Subject: [PATCH] Fall back to base class when adding size distributions. --- cosmic_dustbox/sdist.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cosmic_dustbox/sdist.py b/cosmic_dustbox/sdist.py index a03b381..f2537d6 100644 --- a/cosmic_dustbox/sdist.py +++ b/cosmic_dustbox/sdist.py @@ -128,11 +128,11 @@ class SizeDist(object): elif callable(other): def func(sizes): return self.func(sizes) + other(sizes) - return self.__class__(self.sizeMin, self.sizeMax, func) + return SizeDist(self.sizeMin, self.sizeMax, func) else: def func(sizes): return other + self.func(sizes) - return self.__class__(self.sizeMin, self.sizeMax, func) + return SizeDist(self.sizeMin, self.sizeMax, func) # make addition commutative __radd__ = __add__ -- GitLab