diff --git a/CHANGELOG.md b/CHANGELOG.md index 1b1114d9..b8ab28fe 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -11,6 +11,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ### Changed - Loosened `pandas`, `torch` and `torch-light` versions for `python >= 3.8` ([#58](https://github.com/MobileTeleSystems/RecTools/pull/58)) +### Fixed +- Bug in `Interactions.from_raw` method ([#58](https://github.com/MobileTeleSystems/RecTools/pull/58)) + ## [0.4.2] - 01.12.2023 diff --git a/rectools/dataset/interactions.py b/rectools/dataset/interactions.py index ef412905..3b1ccf52 100644 --- a/rectools/dataset/interactions.py +++ b/rectools/dataset/interactions.py @@ -115,8 +115,8 @@ def from_raw( Columns.Item: item_id_map.convert_to_internal(interactions[Columns.Item]), }, ) - df[Columns.Weight] = interactions[Columns.Weight] - df[Columns.Datetime] = interactions[Columns.Datetime] + df[Columns.Weight] = interactions[Columns.Weight].values + df[Columns.Datetime] = interactions[Columns.Datetime].values cls._convert_weight_and_datetime_types(df) return cls(df)