From 75db3ba43f572a77b60551d54bf00ddc9df91ae6 Mon Sep 17 00:00:00 2001 From: rwijtvliet Date: Wed, 31 Jan 2024 21:41:13 +0100 Subject: [PATCH] forgot aftercare argument --- belvys/tenant.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/belvys/tenant.py b/belvys/tenant.py index 951e806..dd291ba 100644 --- a/belvys/tenant.py +++ b/belvys/tenant.py @@ -260,7 +260,7 @@ def portfolio_pfl( self._fetch_series(ts_tree, ts_left, ts_right, missing2zero=missing2zero) ts_trees.append(ts_tree) # Turn into series. - series_trees = [to_series_tree(ts_tree) for ts_tree in ts_trees] + series_trees = [to_series_tree(ts_tree, self.aftercare) for ts_tree in ts_trees] if output_as_series: return series_trees # Turn into portfolio line. @@ -312,7 +312,10 @@ def to_series_tree(ts_tree: TsTree, aftercare: Aftercare) -> SeriesTree: if isinstance(ts_tree, Dict): # nested ts_dict = ts_tree - return {name: to_series_tree(subtree) for name, subtree in ts_dict.items()} + return { + name: to_series_tree(subtree, aftercare) + for name, subtree in ts_dict.items() + } # FlatPfLine.