set predict slicer updates

casact · Feb 12, 2020 · ac16880 · ac16880
1 parent cfd13a4
commit ac16880
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/chainladder/methods/benktander.py b/chainladder/methods/benktander.py
@@ -84,6 +84,7 @@ def _get_ultimate_(self, X, sample_weight, obj):
         ult.ddims = np.array([None])
         ult.valuation = pd.DatetimeIndex([pd.to_datetime('2262-04-11')] *
                                          ult.shape[origin])
+        ult._set_slicers()
         return ult
 
     def predict(self, X, sample_weight):

diff --git a/chainladder/methods/capecod.py b/chainladder/methods/capecod.py
@@ -96,6 +96,8 @@ def _get_ultimate_(self, X, sample_weight, obj):
         ult.valuation = pd.DatetimeIndex([pd.to_datetime('2262-04-11')] *
                                          len_orig)
         apriori_._set_slicers()
+        ult._set_slicers()
+        detrended_apriori_._set_slicers()
         return ult, apriori_, detrended_apriori_
 
     def predict(self, X, sample_weight):