Skip to content

Commit 16e2ec0

Browse files
committed
Fix: Correctly manage reference window labels and initialize current window variables in ReferenceWindowModel
1 parent 145d6eb commit 16e2ec0

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pysad/models/integrations/reference_window_model.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,12 +75,13 @@ def fit_partial(self, X, y=None):
7575
0, len(self.reference_window_X) - self.window_size):]
7676

7777
if y is not None:
78-
self.reference_window_y = self.reference_window_y[max(
79-
0, len(self.reference_window_y) - self.window_size):]
8078
self.reference_window_y = np.concatenate(
8179
[self.reference_window_y, self.cur_window_y], axis=0)
80+
self.reference_window_y = self.reference_window_y[max(
81+
0, len(self.reference_window_y) - self.window_size):]
8282

83-
self.cur_window = []
83+
self.cur_window_X = []
84+
self.cur_window_y = []
8485
self._fit_model()
8586

8687
return self

0 commit comments

Comments
 (0)