replay-rec 0.20.0__py3-none-any.whl → 0.20.1__py3-none-any.whl

replay/splitters/random_splitter.py

@@ -1,4 +1,4 @@
-from typing import Optional, Tuple
+from typing import Optional
 
 from replay.utils import DataFrameLike, PandasDataFrame, PolarsDataFrame, SparkDataFrame
 
@@ -48,20 +48,20 @@ class RandomSplitter(Splitter):
 
     def _random_split_spark(
         self, interactions: SparkDataFrame, threshold: float
-    ) -> Tuple[SparkDataFrame, SparkDataFrame]:
+    ) -> tuple[SparkDataFrame, SparkDataFrame]:
         train, test = interactions.randomSplit([1 - threshold, threshold], self.seed)
         return train, test
 
     def _random_split_pandas(
         self, interactions: PandasDataFrame, threshold: float
-    ) -> Tuple[PandasDataFrame, PandasDataFrame]:
+    ) -> tuple[PandasDataFrame, PandasDataFrame]:
         train = interactions.sample(frac=(1 - threshold), random_state=self.seed)
         test = interactions.drop(train.index)
         return train, test
 
     def _random_split_polars(
         self, interactions: PolarsDataFrame, threshold: float
-    ) -> Tuple[PolarsDataFrame, PolarsDataFrame]:
+    ) -> tuple[PolarsDataFrame, PolarsDataFrame]:
         train_size = int(len(interactions) * (1 - threshold)) + 1
         shuffled_interactions = interactions.sample(fraction=1, shuffle=True, seed=self.seed)
         train = shuffled_interactions[:train_size]

replay/splitters/ratio_splitter.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, Tuple
+from typing import Optional
 
 import polars as pl
 
@@ -200,7 +200,7 @@ class RatioSplitter(Splitter):
 
     def _partial_split_fractions(
         self, interactions: DataFrameLike, ratio: float
-    ) -> Tuple[DataFrameLike, DataFrameLike]:
+    ) -> tuple[DataFrameLike, DataFrameLike]:
         res = self._add_time_partition(interactions)
         train_size = round(1 - ratio, self._precision)
 
@@ -212,7 +212,7 @@ class RatioSplitter(Splitter):
 
     def _partial_split_fractions_pandas(
         self, interactions: PandasDataFrame, train_size: float
-    ) -> Tuple[PandasDataFrame, PandasDataFrame]:
+    ) -> tuple[PandasDataFrame, PandasDataFrame]:
         interactions["count"] = interactions.groupby(self.divide_column, sort=False)[self.divide_column].transform(len)
         interactions["frac"] = (interactions["row_num"] / interactions["count"]).round(self._precision)
         if self.min_interactions_per_group is not None:
@@ -229,7 +229,7 @@ class RatioSplitter(Splitter):
 
     def _partial_split_fractions_spark(
         self, interactions: SparkDataFrame, train_size: float
-    ) -> Tuple[SparkDataFrame, SparkDataFrame]:
+    ) -> tuple[SparkDataFrame, SparkDataFrame]:
         interactions = interactions.withColumn(
             "count", sf.count(self.timestamp_column).over(Window.partitionBy(self.divide_column))
         )
@@ -257,7 +257,7 @@ class RatioSplitter(Splitter):
 
     def _partial_split_fractions_polars(
         self, interactions: PolarsDataFrame, train_size: float
-    ) -> Tuple[PolarsDataFrame, PolarsDataFrame]:
+    ) -> tuple[PolarsDataFrame, PolarsDataFrame]:
         interactions = interactions.with_columns(
             pl.count(self.timestamp_column).over(pl.col(self.divide_column)).alias("count")
         )
@@ -282,7 +282,7 @@ class RatioSplitter(Splitter):
 
         return train, test
 
-    def _partial_split(self, interactions: DataFrameLike, ratio: float) -> Tuple[DataFrameLike, DataFrameLike]:
+    def _partial_split(self, interactions: DataFrameLike, ratio: float) -> tuple[DataFrameLike, DataFrameLike]:
         res = self._add_time_partition(interactions)
         if isinstance(res, SparkDataFrame):
             return self._partial_split_spark(res, ratio)
@@ -293,7 +293,7 @@ class RatioSplitter(Splitter):
 
     def _partial_split_pandas(
         self, interactions: PandasDataFrame, ratio: float
-    ) -> Tuple[PandasDataFrame, PandasDataFrame]:
+    ) -> tuple[PandasDataFrame, PandasDataFrame]:
         interactions["count"] = interactions.groupby(self.divide_column, sort=False)[self.divide_column].transform(len)
         interactions["train_size"] = interactions["count"] - (interactions["count"] * ratio).astype(int)
         if self.min_interactions_per_group is not None:
@@ -319,7 +319,7 @@ class RatioSplitter(Splitter):
 
         return train, test
 
-    def _partial_split_spark(self, interactions: SparkDataFrame, ratio: float) -> Tuple[SparkDataFrame, SparkDataFrame]:
+    def _partial_split_spark(self, interactions: SparkDataFrame, ratio: float) -> tuple[SparkDataFrame, SparkDataFrame]:
         interactions = interactions.withColumn(
             "count", sf.count(self.timestamp_column).over(Window.partitionBy(self.divide_column))
         )
@@ -352,7 +352,7 @@ class RatioSplitter(Splitter):
 
     def _partial_split_polars(
         self, interactions: PolarsDataFrame, ratio: float
-    ) -> Tuple[PolarsDataFrame, PolarsDataFrame]:
+    ) -> tuple[PolarsDataFrame, PolarsDataFrame]:
         interactions = interactions.with_columns(
             pl.count(self.timestamp_column).over(self.divide_column).alias("count")
         )
@@ -385,7 +385,7 @@ class RatioSplitter(Splitter):
 
         return train, test
 
-    def _core_split(self, interactions: DataFrameLike) -> List[DataFrameLike]:
+    def _core_split(self, interactions: DataFrameLike) -> list[DataFrameLike]:
         if self.split_by_fractions:
             return self._partial_split_fractions(interactions, self.test_size)
         else:

replay/splitters/time_splitter.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import List, Optional, Tuple, Union
+from typing import Optional, Union
 
 import polars as pl
 
@@ -150,7 +150,7 @@ class TimeSplitter(Splitter):
 
     def _partial_split(
         self, interactions: DataFrameLike, threshold: Union[datetime, str, int]
-    ) -> Tuple[DataFrameLike, DataFrameLike]:
+    ) -> tuple[DataFrameLike, DataFrameLike]:
         if isinstance(threshold, str):
             threshold = datetime.strptime(threshold, self.time_column_format)
 
@@ -166,7 +166,7 @@ class TimeSplitter(Splitter):
 
     def _partial_split_pandas(
         self, interactions: PandasDataFrame, threshold: Union[datetime, str, int]
-    ) -> Tuple[PandasDataFrame, PandasDataFrame]:
+    ) -> tuple[PandasDataFrame, PandasDataFrame]:
         res = interactions.copy(deep=True)
         if isinstance(threshold, float):
             res.sort_values(self.timestamp_column, inplace=True)
@@ -186,7 +186,7 @@ class TimeSplitter(Splitter):
 
     def _partial_split_spark(
         self, interactions: SparkDataFrame, threshold: Union[datetime, str, int]
-    ) -> Tuple[SparkDataFrame, SparkDataFrame]:
+    ) -> tuple[SparkDataFrame, SparkDataFrame]:
         if isinstance(threshold, float):
             dates = interactions.select(self.timestamp_column).withColumn(
                 "_row_number_by_ts", sf.row_number().over(Window.orderBy(self.timestamp_column))
@@ -208,7 +208,7 @@ class TimeSplitter(Splitter):
 
     def _partial_split_polars(
         self, interactions: PolarsDataFrame, threshold: Union[datetime, str, int]
-    ) -> Tuple[PolarsDataFrame, PolarsDataFrame]:
+    ) -> tuple[PolarsDataFrame, PolarsDataFrame]:
         if isinstance(threshold, float):
             test_start = int(len(interactions) * (1 - threshold)) + 1
 
@@ -225,5 +225,5 @@ class TimeSplitter(Splitter):
 
         return train, test
 
-    def _core_split(self, interactions: DataFrameLike) -> List[DataFrameLike]:
+    def _core_split(self, interactions: DataFrameLike) -> list[DataFrameLike]:
         return self._partial_split(interactions, self.time_threshold)

replay/splitters/two_stage_splitter.py

@@ -2,7 +2,7 @@
 This splitter split data by two columns.
 """
 
-from typing import Optional, Tuple
+from typing import Optional
 
 import numpy as np
 import polars as pl
@@ -165,7 +165,7 @@ class TwoStageSplitter(Splitter):
 
         return test_users
 
-    def _split_proportion_spark(self, interactions: SparkDataFrame) -> Tuple[SparkDataFrame, SparkDataFrame]:
+    def _split_proportion_spark(self, interactions: SparkDataFrame) -> tuple[SparkDataFrame, SparkDataFrame]:
         counts = interactions.groupBy(self.first_divide_column).count()
         test_users = self._get_test_values(interactions).withColumn("is_test", sf.lit(True))
         if self.shuffle:
@@ -197,7 +197,7 @@ class TwoStageSplitter(Splitter):
 
         return train, test
 
-    def _split_proportion_pandas(self, interactions: PandasDataFrame) -> Tuple[PandasDataFrame, PandasDataFrame]:
+    def _split_proportion_pandas(self, interactions: PandasDataFrame) -> tuple[PandasDataFrame, PandasDataFrame]:
         counts = (
             interactions.groupby(self.first_divide_column).agg(count=(self.first_divide_column, "count")).reset_index()
         )
@@ -224,7 +224,7 @@ class TwoStageSplitter(Splitter):
 
         return train, test
 
-    def _split_proportion_polars(self, interactions: PolarsDataFrame) -> Tuple[PolarsDataFrame, PolarsDataFrame]:
+    def _split_proportion_polars(self, interactions: PolarsDataFrame) -> tuple[PolarsDataFrame, PolarsDataFrame]:
         counts = interactions.group_by(self.first_divide_column).count()
         test_users = self._get_test_values(interactions).with_columns(pl.lit(True).alias("is_test"))
         if self.shuffle:

replay/utils/__init__.py

@@ -1,5 +1,6 @@
 from .types import (
     ANN_AVAILABLE,
+    LIGHTFM_AVAILABLE,
     OPENVINO_AVAILABLE,
     OPTUNA_AVAILABLE,
     PYSPARK_AVAILABLE,
@@ -14,4 +15,3 @@ from .types import (
     PolarsDataFrame,
     SparkDataFrame,
 )
-from .warnings import deprecation_warning

replay/utils/common.py

@@ -73,7 +73,7 @@ def load_from_replay(path: Union[str, Path], **kwargs) -> SavableObject:
     :param path: Path to save the object.
     """
     path = Path(path).with_suffix(".replay").resolve()
-    with open(path / "init_args.json", "r") as file:
+    with open(path / "init_args.json") as file:
         class_name = json.loads(file.read())["_class_name"]
     obj_type = globals()[class_name]
     obj = obj_type.load(path, **kwargs)

replay/utils/session_handler.py

@@ -6,7 +6,7 @@ import logging
 import os
 import sys
 from math import floor
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 import psutil
 
@@ -113,7 +113,7 @@ class Borg:
     This class allows to share objects between instances.
     """
 
-    _shared_state: Dict[str, Any] = {}
+    _shared_state: dict[str, Any] = {}
 
     def __init__(self):
         self.__dict__ = self._shared_state

replay/utils/spark_utils.py

@@ -3,7 +3,8 @@ import logging
 import os
 import pickle
 import warnings
-from typing import Any, Iterable, List, Optional, Tuple, Union
+from collections.abc import Iterable
+from typing import Any, Optional, Union
 
 import numpy as np
 import pandas as pd
@@ -90,7 +91,7 @@ def convert2spark(data_frame: Optional[DataFrameLike]) -> Optional[SparkDataFram
 def get_top_k(
     dataframe: SparkDataFrame,
     partition_by_col: Column,
-    order_by_col: List[Column],
+    order_by_col: list[Column],
     k: int,
 ) -> SparkDataFrame:
     """
@@ -393,7 +394,7 @@ def horizontal_explode(
     data_frame: SparkDataFrame,
     column_to_explode: str,
     prefix: str,
-    other_columns: List[Column],
+    other_columns: list[Column],
 ) -> SparkDataFrame:
     """
     Transform a column with an array of values into separate columns.
@@ -509,7 +510,7 @@ def unpersist_if_exists(dataframe: Optional[SparkDataFrame]) -> None:
 def join_with_col_renaming(
     left: SparkDataFrame,
     right: SparkDataFrame,
-    on_col_name: Union[str, List],
+    on_col_name: Union[str, list],
     how: str = "inner",
     suffix="join",
 ) -> SparkDataFrame:
@@ -698,7 +699,7 @@ def filter_cold(
     df: Optional[SparkDataFrame],
     warm_df: SparkDataFrame,
     col_name: str,
-) -> Tuple[int, Optional[SparkDataFrame]]:
+) -> tuple[int, Optional[SparkDataFrame]]:
     """
     Filter out new user/item ids absent in `warm_df`.
     Return number of new users/items and filtered dataframe.

replay/utils/types.py

@@ -1,5 +1,6 @@
+from collections.abc import Iterable
 from importlib.util import find_spec
-from typing import Iterable, Union
+from typing import Union
 
 from pandas import DataFrame as PandasDataFrame
 from polars import DataFrame as PolarsDataFrame
@@ -46,3 +47,4 @@ ANN_AVAILABLE = all(
 )
 OPENVINO_AVAILABLE = TORCH_AVAILABLE and find_spec("onnx") and find_spec("openvino")
 OPTUNA_AVAILABLE = find_spec("optuna")
+LIGHTFM_AVAILABLE = find_spec("lightfm")

{replay_rec-0.20.0.dist-info → replay_rec-0.20.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: replay-rec
-Version: 0.20.0
+Version: 0.20.1
 Summary: RecSys Library
 License-Expression: Apache-2.0
 License-File: LICENSE
@@ -245,6 +245,12 @@ pip install openvino onnx
 pip install hnswlib fixed-install-nmslib
 ```
 
+4) (Experimental) LightFM model support:
+```bash
+pip install ligfhtfm
+```
+> **_NOTE_** : LightFM is not officially supported for Python 3.12 due to discontinued maintenance of the library. If you wish to install it locally, you'll have to use a patched fork of LightFM, such as the [one used internally](https://github.com/daviddavo/lightfm).
+
 
 <a name="examples"></a>
 ## 📑  Resources

{replay_rec-0.20.0.dist-info → replay_rec-0.20.1.dist-info}/RECORD

@@ -1,45 +1,45 @@
-replay/__init__.py,sha256=F7o-BrLxOoILrOfwnTGOC1UkgPgzG6fimigpkS6Eib0,225
+replay/__init__.py,sha256=lcQXZRcrHeoQ5A4N4ZuFAUfXTICM00LWZ-6_KwdDxmM,225
 replay/data/__init__.py,sha256=g5bKRyF76QL_BqlED-31RnS8pBdcyj9loMsx5vAG_0E,301
-replay/data/dataset.py,sha256=JNKSg92FrlI7065IMgkw92RpGCqRV_y5TcXzhCNFZig,30745
+replay/data/dataset.py,sha256=yQDc8lfphQYfHpm_T1MhnG8_GyM4ONyxJoFc1rUgdJ8,30755
 replay/data/dataset_utils/__init__.py,sha256=9wUvG8ZwGUvuzLU4zQI5FDcH0WVVo5YLN2ey3DterP0,55
-replay/data/dataset_utils/dataset_label_encoder.py,sha256=6Pd6LYXLL_9xwvV5hbJvL88Um-KSIeVeYTa3RopuHP0,9807
+replay/data/dataset_utils/dataset_label_encoder.py,sha256=bxuJPhShFZBok7bQZYGNMV1etCLNTJUpyKO5MIwWack,9823
 replay/data/nn/__init__.py,sha256=nj2Ep-tduuQkc-TnBkvN8-rDnFbcWO2oZrfcXl9M3C8,1122
-replay/data/nn/schema.py,sha256=0I3fdMYuMElcAhVfY44eFDGq1tVYRUMvZBspdVUqyfI,17112
-replay/data/nn/sequence_tokenizer.py,sha256=vVOoSxh8FaTaL6owVWyKZzOtY91yxUYNdQFmwOLGvJE,37270
-replay/data/nn/sequential_dataset.py,sha256=rUEjcA5j5wG_UpN_BxoqgeLSiJFPyxPrJEI-futqzro,11161
-replay/data/nn/torch_sequential_dataset.py,sha256=vXOsUAjj5_N6OmQ0grZhqfbiqFLPjNs8lALyjPEbL-M,11546
-replay/data/nn/utils.py,sha256=44E_jyFtpYonVj9rQeRGhehF91oe1DP2QOSfsoemiTQ,3270
-replay/data/schema.py,sha256=iQormeeiLbbxyADdV6Dv_KENe-tIaRiqlhixkQlC2aw,15952
+replay/data/nn/schema.py,sha256=h1KgaNV-hgN9Vpt24c92EmeMpm_8W0s9a2M0wLxJHYk,17101
+replay/data/nn/sequence_tokenizer.py,sha256=_9fBF-84jdn8Pa3pFKIr6prUjNYCc6BVzwRl9VSleKQ,37419
+replay/data/nn/sequential_dataset.py,sha256=qthp87SQ44VpgoH3RKsqm6CxCeQyApn58l7_16txAZM,11303
+replay/data/nn/torch_sequential_dataset.py,sha256=QSh4IM2vzAF095_ZMC1gMqZj9slHXos9gfx_R_DlpGM,11545
+replay/data/nn/utils.py,sha256=Ic3G4yZRIzBYXLmwP1VstlZXPNR7AYGCc5EyZAERp5c,3297
+replay/data/schema.py,sha256=JmYLCrNgBS5oq4O_PT724Gr1pDurHEykcqV8Xaj0XTw,15922
 replay/data/spark_schema.py,sha256=4o0Kn_fjwz2-9dBY3q46F9PL0F3E7jdVpIlX7SG3OZI,1111
 replay/metrics/__init__.py,sha256=j0PGvUehaPEZMNo9SQwJsnvzrS4bam9eHrRMQFLnMjY,2813
-replay/metrics/base_metric.py,sha256=8A7vQR3FimW0w3cwJXnpruuL7VhdPhUs1LRUs6wAdig,15995
-replay/metrics/categorical_diversity.py,sha256=OYsF-Ng-WrF9CC-sKgQKngrA779NO8MtgRvvAyC8MXM,10781
-replay/metrics/coverage.py,sha256=4j73A1-d7c4GynOSmfUveg--pBhIwaDBcWQABFeHL-Y,8540
+replay/metrics/base_metric.py,sha256=ejtwFHktN4J8Fi1HIM3w0zlMAd8nO7-XpFi2D1iHXUQ,16010
+replay/metrics/categorical_diversity.py,sha256=3tp8n457Ob4gjM-UTB5N19u9WAF7fLDkWKk-Mth-Vzc,10769
+replay/metrics/coverage.py,sha256=e6vPItrRlI-mLNuOT5uoo5lMAAzkYGKZRxvupi21dMk,8528
 replay/metrics/descriptors.py,sha256=BHORyGKfJgPeUjgLO0u2urSTe16UQbb-HHh8soqnwDE,3893
-replay/metrics/experiment.py,sha256=oUGFZoCbdBtAqNiQ8ll3FxJ94r5ts_Tn8snZLTAeYcY,8137
-replay/metrics/hitrate.py,sha256=G0_GufGc1RIQ_RntCIvDn0piJIVyT3W35bTN-Un9-Ug,2339
-replay/metrics/map.py,sha256=qUPFkyrvRf7gEyxosBApHOxpAVZJ3Yksqe4vsbzTukE,2597
-replay/metrics/mrr.py,sha256=AkfRdvL21Ri1zE4Cmib2twJaEaeBO5wze87eZ4lkYlc,2188
-replay/metrics/ndcg.py,sha256=wzdFFf9rmG6yH7klK5V9ab1wNdI6CHXezvYasxO9_ZU,3134
-replay/metrics/novelty.py,sha256=4S6PsdOl8lqweBlemWJh2fVqMiN8jfj3cIU-9TxZXLY,5500
-replay/metrics/offline_metrics.py,sha256=wSK8xrSlnCVaCBEPZKPgbR2mOQISp_GMZEbIao3u9wI,20473
-replay/metrics/precision.py,sha256=8hfz9bLZt7wUfyJtJPiIoDEOP_bweZSpCS6Z1Cm9IGE,2281
-replay/metrics/recall.py,sha256=ATN5OIeFKMG7-nqpPCBPLItrUhGRucLCELWnzRyCRoo,2472
-replay/metrics/rocauc.py,sha256=gybFcyIyVDpwoR9K38nOSxlAjIYbFqXUzABuckxqHps,3307
-replay/metrics/surprisal.py,sha256=mcGOQTvlNcGz4vNi-U-qe-CJqlYICKMRny5KtJVMKAc,7538
-replay/metrics/torch_metrics_builder.py,sha256=2gcCcb0A-TVpYcBIYGhXrggyFX-M_T7Q1pQUiMpxEZE,13845
-replay/metrics/unexpectedness.py,sha256=cfDnkpK6nPeawwHDVNQAkUtsW0SvAttI84k4M5ttkyo,6888
+replay/metrics/experiment.py,sha256=6Sw8PyItn3E2R-BBa_YwrmtBV3n0uAGHHOvkhHYgMz4,8125
+replay/metrics/hitrate.py,sha256=LcOJLMs3_Dq4_pbKx95qdCdjGrX52dyWyuWUFXCyaDw,2314
+replay/metrics/map.py,sha256=dIZcmUxd2XnNC7d_d7gmq0cjNaI1hlNMaJTSHGCokQE,2572
+replay/metrics/mrr.py,sha256=qM8tVMSoyYR-kTx0mnBGppoC53SxNlZKm7JKMUmSv9U,2163
+replay/metrics/ndcg.py,sha256=izajmD243ZIK3KLm9M-NtLwxb9N3Ktj58__AAfwF6Vc,3110
+replay/metrics/novelty.py,sha256=j3p1fbUVi2QQgEre42jeQx73PYYDUhy5gYlrL4BL5b8,5488
+replay/metrics/offline_metrics.py,sha256=f_U4Tk3Ke5sR0_OYvoE2_nD6wrOCveg3DM3B9pStVUI,20454
+replay/metrics/precision.py,sha256=DRlsgY_b4bJCOSZjCA58N41REMiDt-dbagRSXxfXyvY,2256
+replay/metrics/recall.py,sha256=fzpASDiH88zcpXJZTbStQ3nuzzSdhd9k1wjF27rM4wc,2447
+replay/metrics/rocauc.py,sha256=1vaVEK7DQTL8BX-i7A64hTFWyO38aNycscPGrdWKwbA,3282
+replay/metrics/surprisal.py,sha256=HkmYrOuw3jydxFrkidjdcpAcKz2DeOnMsKqwB2g9pwY,7526
+replay/metrics/torch_metrics_builder.py,sha256=jccdTNXJrwiWmBoD9cB3ilIn-upKhR6toAfKTG5T2Mc,13855
+replay/metrics/unexpectedness.py,sha256=LSi-z50l3_yrvLnmToHQzm6Ygf2QpNt_zhk6jdg7QUo,6882
 replay/models/__init__.py,sha256=kECYluQZ83zRUWaHVvnt7Tg3BerHrJy9v8XfRxsqyYY,1123
-replay/models/als.py,sha256=eGiMok_zu5ZUKXU9i9feCP4RGMqSnlIGHjks6MqKzHw,6227
-replay/models/association_rules.py,sha256=At0Tw0nhxQno6-UJWqA8vQWF-CgDyWNPcy8TtMhUXhk,14492
-replay/models/base_neighbour_rec.py,sha256=yWqpM7-w8HfmaLDFs7Yt5EZxQdKEZk6mBbEl0sNBJe0,7762
-replay/models/base_rec.py,sha256=grXuXtrOy4BA15TZUMRM4XU8aa0VxJVSdmdvWVMMsDA,56059
-replay/models/cat_pop_rec.py,sha256=K0CF2A4cHzTv9BJ9hQevarrvwKJnHlUpjhemC4de2_w,11941
+replay/models/als.py,sha256=1MFAbcx64tv0MX1wE9CM1NxKD3F3ZDhZUrmt6dvHu74,6220
+replay/models/association_rules.py,sha256=shBNsKjlii0YK-XA6bSl5Ov0ZFTnjxZbgKJU9PFYptY,14507
+replay/models/base_neighbour_rec.py,sha256=SdGb2ejpYjHmxFNTk5zwEo0RWdfPAj1vKGP_oj7IrQo,7783
+replay/models/base_rec.py,sha256=aNIEbSy8G5q92NOpDlSJbp0Z-lAkazFLa9eDAajl1wI,56067
+replay/models/cat_pop_rec.py,sha256=ed1X1PDQY41hFJ1cO3Q5OWy0rXhV5_n23hJ-QHWONtE,11968
 replay/models/cluster.py,sha256=9JcpGnbfgFa4UsyxPAa4WMuJFa3rsuAxiKoy-s_UfyE,4970
 replay/models/common.py,sha256=rFmfwwzWCWED2HaDVuSN7ZUAgaNPGPawUudgn4IApbo,2121
 replay/models/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 replay/models/extensions/ann/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-replay/models/extensions/ann/ann_mixin.py,sha256=SRsj3-zqXNpLQ7ukY0eNCwk0HDKpaKSdIhozRTCqQM0,7755
+replay/models/extensions/ann/ann_mixin.py,sha256=Ua1fuwrvtISNDQ8iPV-ln8S1LDKz8-rIU2UYsMExAiU,7782
 replay/models/extensions/ann/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 replay/models/extensions/ann/entities/base_hnsw_param.py,sha256=5GRdcQj4-zhNXfJ7ko2WHGHgRuXCzSHCRcRxljl1V4c,776
 replay/models/extensions/ann/entities/hnswlib_param.py,sha256=j3V4JXM_yfR6s2TjYakIXMg-zS1-MrP6an930DEIWGM,2104
@@ -48,8 +48,8 @@ replay/models/extensions/ann/index_builders/__init__.py,sha256=47DEQpj8HBSa-_TIm
 replay/models/extensions/ann/index_builders/base_index_builder.py,sha256=Ul25G0FaNLOXUjrDXxZDTg7tLXlv1N6wR8kWjWICtZ0,2110
 replay/models/extensions/ann/index_builders/driver_hnswlib_index_builder.py,sha256=U8-3lRahyWmWkZ7tYuO-Avd1jX-lGh7JukC140wJ-WQ,1600
 replay/models/extensions/ann/index_builders/driver_nmslib_index_builder.py,sha256=1NLWyAJGYgp46uUBhUYQyd0stmG6DhLh7U4JEne5TFw,1308
-replay/models/extensions/ann/index_builders/executor_hnswlib_index_builder.py,sha256=_JZfewcLsU3sRR95_FPlgLfn9vez0JXMxer_024UHK4,2449
-replay/models/extensions/ann/index_builders/executor_nmslib_index_builder.py,sha256=_TClG_czFb50dXyUjjXL4HE5j_i4BH011wjzDGba2U8,2591
+replay/models/extensions/ann/index_builders/executor_hnswlib_index_builder.py,sha256=cf3LhBCRRN-lBYGlJbv8vnY-KVeHAleN5cVjvd58Ibs,2476
+replay/models/extensions/ann/index_builders/executor_nmslib_index_builder.py,sha256=0DPJ3WAt0cZ5dmtZv87fmMEgYXWf8rM35f7CA_DgWZY,2618
 replay/models/extensions/ann/index_builders/nmslib_index_builder_mixin.py,sha256=AIkVnobesnTM5lrBSWf9gd0CySwFQ0vH_DjemfLS4Cs,1925
 replay/models/extensions/ann/index_inferers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 replay/models/extensions/ann/index_inferers/base_inferer.py,sha256=I39aqEc2somfndrCd-KC3XYZnYSrJ2hGpR9y6wO93NA,2524
@@ -67,73 +67,72 @@ replay/models/extensions/ann/index_stores/utils.py,sha256=6r2GP_EFCaCguolW857pb4
 replay/models/extensions/ann/utils.py,sha256=AgQvThi_DvEtakQeTno9hVZVWiWMFHKTjRcQ2wLa5vk,1222
 replay/models/kl_ucb.py,sha256=L6vC2KsTBTTx4ckmGhWybOiLa5Wt54N7cgl7jS2FQRg,6731
 replay/models/knn.py,sha256=HEiGHHQg9pV1_EIWZHfK-XD0BNAm1bj1c0ND9rYnj3k,8992
-replay/models/lin_ucb.py,sha256=Je1FfW-bRFSBE45XIXxQvzJr8sZQuMzIkCs-pb_D8po,16922
+replay/models/lin_ucb.py,sha256=iAR3PbbaQKqmisOKEx9ZyfpxnxcZomr6YauG4mvSakU,18800
 replay/models/nn/__init__.py,sha256=AT3o1qXaxUq4_QIGlcGuSs54ZpueOo-SbpZwuGI-6os,41
 replay/models/nn/loss/__init__.py,sha256=s3iO9QTZvLz_ony2b5K0hEmDmitrXQnAe9j6BRxLpR4,53
 replay/models/nn/loss/sce.py,sha256=p6LFtoYSY4j2pQh6Z7i6cEADCmRnvTgnb8EJXseRKKg,5637
 replay/models/nn/optimizer_utils/__init__.py,sha256=8MHln7CW54oACVUFKdZLjAf4bY83GcOMXpmL87gTnhI,178
-replay/models/nn/optimizer_utils/optimizer_factory.py,sha256=EXqS_2ZxT7K_kFzcjgFpNhmzUoSftf_CwZIIHNHZBgU,2842
+replay/models/nn/optimizer_utils/optimizer_factory.py,sha256=1wicKnya2xrwDaHhqygy1VqB8-3jPDhMM7zY2TJE4dY,2844
 replay/models/nn/sequential/__init__.py,sha256=CI2n0cxs_amqJrwBMq6n0Z_uBOu7CGXfagqvE4Jlmjw,128
 replay/models/nn/sequential/bert4rec/__init__.py,sha256=JfZqHOGxcvOkICl5cWmZbZhaKXpkIvua-Wj57VWWEhw,399
-replay/models/nn/sequential/bert4rec/dataset.py,sha256=VqsXpo7VtVCbV0DqpdTxWnbn76f28nex6SSARQdr_lc,11000
-replay/models/nn/sequential/bert4rec/lightning.py,sha256=g3LSJiIzTnl9nfCEJKRdP2NhLP25b5AnivfSrOeqceU,26559
-replay/models/nn/sequential/bert4rec/model.py,sha256=WSo0l2TVDqlB3_qOfrDWXS7KNOzHgJ6myPv5vazHTlU,17394
+replay/models/nn/sequential/bert4rec/dataset.py,sha256=xd5a-yn5I280Vwoy_KtasDjrvksFolJYp71nDEHNUNQ,10414
+replay/models/nn/sequential/bert4rec/lightning.py,sha256=_hP6_6E1SpGu6b_kiYEF4ZVhwKJ4sj_iPTo6loIvM0o,26546
+replay/models/nn/sequential/bert4rec/model.py,sha256=2Lqvfz7UBB_ArqNs92OD5dy4a1onR4S5dNZiMbZgAgk,17388
 replay/models/nn/sequential/callbacks/__init__.py,sha256=Q7mSZ_RB6iyD7QZaBL_NJ0uh8cRfgxq7gtPHbkSyhoo,282
-replay/models/nn/sequential/callbacks/prediction_callbacks.py,sha256=c9YYhEpfL9D-VPpAHOY_S_6sZcIn1Rtb9G_AZyv3SFY,9309
-replay/models/nn/sequential/callbacks/validation_callback.py,sha256=6TNl3NN9oahK1J7DT44461xqBuUCblCsLzUi2svlhF4,5825
+replay/models/nn/sequential/callbacks/prediction_callbacks.py,sha256=lkYoTOpyt-Gy9wOgOgGlFshTJP24VsDcWm_6pY5Xmyg,9296
+replay/models/nn/sequential/callbacks/validation_callback.py,sha256=GcRWM_yVwRBRLTW1sYCy9_aNZ5C71hmJeGG61Yyh4vA,5812
 replay/models/nn/sequential/compiled/__init__.py,sha256=eSVcCaUH5cDJQRbC7K99X7uMNR-Z-KR4TmYOGKWWJCI,531
-replay/models/nn/sequential/compiled/base_compiled_model.py,sha256=7vS8bumKJu4iyzn5-pqnB_lkPYaZRhaVOVhcHM7Ki_8,10250
+replay/models/nn/sequential/compiled/base_compiled_model.py,sha256=vOL-9jodvSNc7N32V4lTjRTCNM-tOPAxfMyZPENqsFA,10231
 replay/models/nn/sequential/compiled/bert4rec_compiled.py,sha256=Z6nfmdT70Wi-j7_CDFJ88iNCp1gdQleg1WkfHp0hb4s,6400
 replay/models/nn/sequential/compiled/sasrec_compiled.py,sha256=qUaAwQOsBCstOG3RBlj_pJpD8BHmCpLZWCiPBlFVvT4,5856
 replay/models/nn/sequential/postprocessors/__init__.py,sha256=89LGzkNHukcuC2-rfpiz7vmv1zyk6MNY-8zaXrvtn0M,164
-replay/models/nn/sequential/postprocessors/_base.py,sha256=Z-HuXM4WBQtrq9IeCL00jtdb_UopksInxVVK8lxD_hA,1136
-replay/models/nn/sequential/postprocessors/postprocessors.py,sha256=t96EkNVaG560m5BA5FejJKudpUMcfZWc4KrOD12GME4,7820
+replay/models/nn/sequential/postprocessors/_base.py,sha256=Q_SIYKG8G3U03IEK1dtlW1zJI300pOcWQYuMpkY0_nc,1111
+replay/models/nn/sequential/postprocessors/postprocessors.py,sha256=oijLByxuzegVmWZS-qRVhdO7ihqHer6SSGTFa8zX7I8,7810
 replay/models/nn/sequential/sasrec/__init__.py,sha256=c6130lRpPkcbuGgkM7slagBIgH7Uk5zUtSzFDEwAsik,250
-replay/models/nn/sequential/sasrec/dataset.py,sha256=2QpWsXnyZiODSuCesQp3GALA4GsSkwSQQnkTyygdOnU,7768
-replay/models/nn/sequential/sasrec/lightning.py,sha256=1FeadX68S-61Zb4OHP4u8CfMdAKAL49P66gjsChJSTY,25073
-replay/models/nn/sequential/sasrec/model.py,sha256=ZK2_nXtsP5NCBpUXXxKbK0tmT1niMGXlbZZVREUfORU,27669
+replay/models/nn/sequential/sasrec/dataset.py,sha256=L_LeRWqPc__390j8NWVskboS0NqbveIkLwFclcB4oDw,7189
+replay/models/nn/sequential/sasrec/lightning.py,sha256=oScUyB8RU8N4MqWe6kAoWG0JW6Tkb2ldG_jdGFZgA7A,25060
+replay/models/nn/sequential/sasrec/model.py,sha256=8kFovyPWqgQ0hmD3gckRjW7-hLBerl3bgYXCk4PYn0o,27656
 replay/models/optimization/__init__.py,sha256=N8xCuzu0jQGwHrIBjuTRf-ZcZuBJ6FB0d9C5a7izJQU,338
 replay/models/optimization/optuna_mixin.py,sha256=pKu-Vw9l2LsDycubpdJiLkC1eE4pKrDG0T2lhUgRUB4,11960
 replay/models/optimization/optuna_objective.py,sha256=UHWOJwBngPA3IRz9yAMEWPg00oyb7Wq9PXuRPYHIiLE,7538
 replay/models/pop_rec.py,sha256=Ju9y2rU2vW_jFU9-W15fbbr5_ZzYGihSjSxsqKsAf0Q,4964
 replay/models/query_pop_rec.py,sha256=UNsHtf3eQpJom73ZmEO5us4guI4SnCLJYTfuUpRgqes,4086
 replay/models/random_rec.py,sha256=9SC012_X3sNzrAjDG1CPGhjisZb6gnv4VCW7yIMSNpk,8066
-replay/models/slim.py,sha256=CG3_uVT8UdkqGZzkgyTvnrWVWpOtPNbY-2Ec7stfZKg,4557
+replay/models/slim.py,sha256=OAdTS64bObZujzHkq8vfP1kkoLMSWxk1KLg6lCCA0N8,4551
 replay/models/thompson_sampling.py,sha256=gcjlVl1mPiEVt70y8frA762O-eCZzd3SVg1lnDRCEHk,1939
-replay/models/ucb.py,sha256=X98ulD8L3gWR3VA7rbQkXFqQyzWc-Nt12lp_gbLTfLQ,6964
+replay/models/ucb.py,sha256=b2qFgvOAZcyv5triPk18duqF_jt-ty7mypenjRLNWwQ,6952
 replay/models/wilson.py,sha256=o7aUWjq3648dAfgGBoWD5Gu-HzdyobPMaH2lzCLijiA,4558
-replay/models/word2vec.py,sha256=EmrqqPruSUvkWiEtuR6-UynikSRq3yi07CIBerTfqo8,8888
+replay/models/word2vec.py,sha256=atfj6GjR_L-TdurRFr1yi7B3BicJ3ZdFxixW9RfojJs,8882
 replay/preprocessing/__init__.py,sha256=c6wFPAc6lATyp0lE-ZDjHMsXyEMPKX7Usuqylv6H5XQ,597
 replay/preprocessing/converter.py,sha256=JQ-4u5x0eXtswl1iH-bZITBXQov1nebnZ6XcvpD8Twk,4417
-replay/preprocessing/discretizer.py,sha256=VetqOLence4luiLa4HNHBezx3rQViqJTMB9w6MKtSw4,27029
-replay/preprocessing/filters.py,sha256=o-XWUTv4NsciGLwOoYKjVrGrBuOgxTk0c0aY1_tuzcw,45852
-replay/preprocessing/history_based_fp.py,sha256=dWSW2uHMzkqHZQ8J5XckGhYW7T1aR9eznJIruVdtVW8,18714
-replay/preprocessing/label_encoder.py,sha256=XOCL9GwBVd9LUod9sppyTkH-J7TP3oiv4fd9B6rnwmM,41432
+replay/preprocessing/discretizer.py,sha256=jzYqvoSVmiL-oS-ri9Om0vSDoU8bCQimjUoe7FiPfLU,27024
+replay/preprocessing/filters.py,sha256=C0zR4LOnGJsMzowuWfaTPR457RppgLZRhcZFV1WkS7o,45845
+replay/preprocessing/history_based_fp.py,sha256=oEu1CkCz7xcGbPdSTHfhTe1NimnFo50Arn8qngRBgE8,18702
+replay/preprocessing/label_encoder.py,sha256=eWsPa5mZq7_9SDxkaiI8mpCfIKTKNr-tlNmfqEunnTk,41432
 replay/preprocessing/sessionizer.py,sha256=G6i0K3FwqtweRxvcSYraJ-tBWAT2HnV-bWHHlIZJF-s,12217
 replay/scenarios/__init__.py,sha256=XXAKEQPTLlve-0O6NPwFgahFrb4oGcIq3HaYaaGxG2E,94
-replay/scenarios/fallback.py,sha256=EeBmIR-5igzKR2m55bQRFyhxTkpJez6ZkCW449n8hWs,7130
+replay/scenarios/fallback.py,sha256=dO3s9jqYup4rbgMaY6Z6HGm1r7SXkm7jOvNZDr5zm_U,7138
 replay/splitters/__init__.py,sha256=DnqVMelrzLwR8fGQgcWN_8FipGs8T4XGSPOMW-L_x2g,454
-replay/splitters/base_splitter.py,sha256=hj9_GYDWllzv3XnxN6WHu1JKRRVjXo77vZEOLbF9v-s,7761
-replay/splitters/cold_user_random_splitter.py,sha256=i695EVggJfc2a_qOYIa3YxEquJiQINEizYV_TYFLxDo,4430
-replay/splitters/k_folds.py,sha256=WH02_DP18A2ae893ysonmfLPB56_i1ETllTAwaCYekg,6218
-replay/splitters/last_n_splitter.py,sha256=hPHgy1cwiz45UwSu6QAwsLITYhVRJdjbMzkfdiBff-k,15381
-replay/splitters/new_users_splitter.py,sha256=bv_QCPkL7KFxJIovAXQbP3Rlty3My48YNTqrj-2ucFQ,9167
-replay/splitters/random_splitter.py,sha256=mbOcxeF0B9WQ9OSxK8CHkPtO8UzKCZJt3rRyFhn-hyQ,2996
-replay/splitters/ratio_splitter.py,sha256=8zvuCn16Icc4ntQPKXJ5ArAWuJzCZ9NHZtgWctKyBVY,17519
-replay/splitters/time_splitter.py,sha256=tsoK3Qg_pcYHDxBlv2xC8ohAikoIqac3fRGBvCb-QRo,9026
-replay/splitters/two_stage_splitter.py,sha256=U90l1wfJnMAAW1j4YpJSd8zWvWB-LDUKFCifnanXraU,17830
-replay/utils/__init__.py,sha256=8Z-GPkn599thC2rvNQbMsf6mq8949N9lj7HzJUvkMfQ,362
-replay/utils/common.py,sha256=9wbXbAAbqQq2BEYuZEvpasENeDRKf2bmePjH4X24mj8,5363
+replay/splitters/base_splitter.py,sha256=zvYVEHBYrK8Y2qPv3kYachfLFwR9-kUAiU1UJSNGS8A,7749
+replay/splitters/cold_user_random_splitter.py,sha256=32VgAHiwk9Emkofu1KqwGZrrFiyrYtSQ3YPdt5p_XoQ,4423
+replay/splitters/k_folds.py,sha256=RDDL3gE6M5qfK5Ig-pxxJeq3O4uxsWJjLFQRRzQ2Ssg,6211
+replay/splitters/last_n_splitter.py,sha256=hMWIGYFg17LioT08VBXut5Ic-w9oXsKd739cy2xuwYs,15368
+replay/splitters/new_users_splitter.py,sha256=NksAdl_wL9zwHj3cY5NqrrnkOajgyUDloSsRZ9HUE48,9160
+replay/splitters/random_splitter.py,sha256=0DO0qulT0jp_GXswmFh3BMJ7utS-z9e-r5jIrmTKGC4,2989
+replay/splitters/ratio_splitter.py,sha256=rFWN-nKBYx1qKrmtYzjYf08DWFiKOCo5ZRUz-NHJFfs,17506
+replay/splitters/time_splitter.py,sha256=0ZAMK26b--1wjrfzCuNVBh7gMPTa8SGf4LMEgACiUxA,9013
+replay/splitters/two_stage_splitter.py,sha256=8Zn6BTJmZg04CD4l2jmil2dEu6xtglJaSS5mkotIXRc,17823
+replay/utils/__init__.py,sha256=3Skc9bUISEPPMMxdUCCT_S1q-i7cAT3KT0nExe-VMrw,343
+replay/utils/common.py,sha256=92MTG51WpeEQJ2gu-WvdNe4Fmqm8ze-y1VNIAHW81jQ,5358
 replay/utils/dataframe_bucketizer.py,sha256=LipmBBQkdkLGroZpbP9i7qvTombLdMxo2dUUys1m5OY,3748
 replay/utils/distributions.py,sha256=UuhaC9HI6HnUXW97fEd-TsyDk4JT8t7k1T_6l5FpOMs,1203
 replay/utils/model_handler.py,sha256=6WRyd39B-UXTtKTHWD_ssYN1vMmkjd417bwKb50uqJY,5754
-replay/utils/session_handler.py,sha256=0-TB33M34slINnFgOI1eB4Z5LUqKuV5kXmFrC8TldFQ,5144
-replay/utils/spark_utils.py,sha256=62TGUGO0JQHFOLUs8hwdYYSSuypCoYQni-UgTZbV8Ko,27397
+replay/utils/session_handler.py,sha256=fQo2wseow8yuzKnEXT-aYAXcQIgRbTTXp0v7g1VVi0w,5138
+replay/utils/spark_utils.py,sha256=GbRp-MuUoO3Pc4chFvlmo9FskSlRLeNlC3Go5pEJ6Ok,27411
 replay/utils/time.py,sha256=J8asoQBytPcNw-BLGADYIsKeWhIoN1H5hKiX9t2AMqo,9376
-replay/utils/types.py,sha256=bcFoWywKfFcD3DkzI8OZ_gZvOqDpuvnxM3_9P2ZTTHw,1192
-replay/utils/warnings.py,sha256=ByAEhT1jKvq02llCtAXM-ioutLsBZjtiqIBesoVi4_Y,929
-replay_rec-0.20.0.dist-info/METADATA,sha256=8utYsWIybjfpEu1205q_mXaYufsN8CR4hAabe2Eby_8,13218
-replay_rec-0.20.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-replay_rec-0.20.0.dist-info/licenses/LICENSE,sha256=rPmcA7UrHxBChEAAlJyE24qUWKKl9yLQXxFsKeg_LX4,11344
-replay_rec-0.20.0.dist-info/licenses/NOTICE,sha256=k0bo4KHiHLRax5K3XKTTrf2Fi8V91mJ-R3FMdh6Reg0,2002
-replay_rec-0.20.0.dist-info/RECORD,,
+replay/utils/types.py,sha256=rD9q9CqEXgF4yy512Hv2nXclvwcnfodOnhBZ1HSUI4c,1260
+replay_rec-0.20.1.dist-info/METADATA,sha256=gHDb26sYFSRyjHopOesLiM_Dtvf4b36mXdpULv2vx8o,13562
+replay_rec-0.20.1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+replay_rec-0.20.1.dist-info/licenses/LICENSE,sha256=rPmcA7UrHxBChEAAlJyE24qUWKKl9yLQXxFsKeg_LX4,11344
+replay_rec-0.20.1.dist-info/licenses/NOTICE,sha256=k0bo4KHiHLRax5K3XKTTrf2Fi8V91mJ-R3FMdh6Reg0,2002
+replay_rec-0.20.1.dist-info/RECORD,,

replay/utils/warnings.py

@@ -1,26 +0,0 @@
-import functools
-import warnings
-from collections.abc import Callable
-from typing import Any, Optional
-
-
-def deprecation_warning(message: Optional[str] = None) -> Callable[..., Any]:
-    """
-    Decorator that throws deprecation warnings.
-
-    :param message: message to deprecation warning without func name.
-    """
-    base_msg = "will be deprecated in future versions."
-
-    def decorator(func: Callable[..., Any]) -> Callable[..., Any]:
-        @functools.wraps(func)
-        def wrapper(*args: Any, **kwargs: Any) -> Any:
-            msg = f"{func.__qualname__} {message if message else base_msg}"
-            warnings.simplefilter("always", DeprecationWarning)  # turn off filter
-            warnings.warn(msg, category=DeprecationWarning, stacklevel=2)
-            warnings.simplefilter("default", DeprecationWarning)  # reset filter
-            return func(*args, **kwargs)
-
-        return wrapper
-
-    return decorator