fixed bug in Interactions.from_raw

MobileTeleSystems · Dec 3, 2023 · 897674b · 897674b
1 parent 9e11b63
commit 897674b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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)