Skip to content

Commit e61a49e

Browse files
committed
hotfix: nhits adjust
1 parent 0dde958 commit e61a49e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

lightwood/mixer/nhits.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,8 @@ def _make_initial_df(self, df, mode='inference'):
216216
sdf = sdf[sdf['ds'].gt(self.group_boundaries[group])]
217217
if sdf.shape[0] > 0:
218218
filtered.append(sdf)
219-
Y_df = pd.concat(filtered)
219+
if filtered:
220+
Y_df = pd.concat(filtered)
220221

221222
return Y_df
222223

0 commit comments

Comments
 (0)