Skip to content

Commit 06c53fb

Browse files
blonderedfeldlime
andauthored
Update rectools/dataset/interactions.py
Co-authored-by: Emiliy Feldman <[email protected]>
1 parent d1d5ef5 commit 06c53fb

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

rectools/dataset/interactions.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,8 +202,7 @@ def to_external(
202202
if include_datetime:
203203
cols_to_add.append(Columns.Datetime)
204204
if include_extra_cols:
205-
cols_not_to_add = [Columns.User, Columns.Item, Columns.Weight, Columns.Datetime]
206-
extra_cols = [col for col in self.df if col not in cols_not_to_add]
205+
extra_cols = [col for col in self.df if col not in Columns.Interactions]
207206
cols_to_add.extend(extra_cols)
208207

209208
for col in cols_to_add:

0 commit comments

Comments
 (0)