replay-rec 0.17.1rc0__tar.gz → 0.18.0rc0__tar.gz

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.1
 Name: replay-rec
-Version: 0.17.1rc0
+Version: 0.18.0rc0
 Summary: RecSys Library
 Home-page: https://sb-ai-lab.github.io/RePlay/
 License: Apache-2.0
 Author: AI Lab
-Requires-Python: >=3.8.1,<3.11
+Requires-Python: >=3.8.1,<3.12
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers
@@ -16,32 +16,34 @@ Classifier: Operating System :: Unix
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Provides-Extra: all
 Provides-Extra: spark
 Provides-Extra: torch
 Requires-Dist: d3rlpy (>=2.0.4,<3.0.0)
+Requires-Dist: fixed-install-nmslib (==2.1.2)
 Requires-Dist: gym (>=0.26.0,<0.27.0)
-Requires-Dist: hnswlib (==0.7.0)
+Requires-Dist: hnswlib (>=0.7.0,<0.8.0)
 Requires-Dist: implicit (>=0.7.0,<0.8.0)
 Requires-Dist: lightautoml (>=0.3.1,<0.4.0)
 Requires-Dist: lightfm (==1.17)
-Requires-Dist: lightning (>=2.0.2,<3.0.0) ; extra == "torch" or extra == "all"
+Requires-Dist: lightning (>=2.0.2,<=2.4.0) ; extra == "torch" or extra == "all"
 Requires-Dist: llvmlite (>=0.32.1)
-Requires-Dist: nmslib (==2.1.1)
 Requires-Dist: numba (>=0.50)
 Requires-Dist: numpy (>=1.20.0)
 Requires-Dist: optuna (>=3.2.0,<3.3.0)
 Requires-Dist: pandas (>=1.3.5,<=2.2.2)
-Requires-Dist: polars (>=0.20.7,<0.21.0)
-Requires-Dist: psutil (>=5.9.5,<5.10.0)
+Requires-Dist: polars (>=1.0.0,<1.1.0)
+Requires-Dist: psutil (>=6.0.0,<6.1.0)
 Requires-Dist: pyarrow (>=12.0.1)
-Requires-Dist: pyspark (>=3.0,<3.5) ; extra == "spark" or extra == "all"
+Requires-Dist: pyspark (>=3.0,<3.5) ; (python_full_version >= "3.8.1" and python_version < "3.11") and (extra == "spark" or extra == "all")
+Requires-Dist: pyspark (>=3.4,<3.5) ; (python_version >= "3.11" and python_version < "3.12") and (extra == "spark" or extra == "all")
 Requires-Dist: pytorch-ranger (>=0.1.1,<0.2.0) ; extra == "torch" or extra == "all"
-Requires-Dist: sb-obp (>=0.5.7,<0.6.0)
+Requires-Dist: sb-obp (>=0.5.8,<0.6.0)
 Requires-Dist: scikit-learn (>=1.0.2,<2.0.0)
-Requires-Dist: scipy (>=1.8.1,<1.9.0)
-Requires-Dist: torch (>=1.8,<2.0) ; extra == "torch" or extra == "all"
+Requires-Dist: scipy (>=1.8.1,<2.0.0)
+Requires-Dist: torch (>=1.8,<=2.4.0) ; extra == "torch" or extra == "all"
 Project-URL: Repository, https://github.com/sb-ai-lab/RePlay
 Description-Content-Type: text/markdown
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "poetry_dynamic_versioning.backend"
 
 [tool.black]
 line-length = 120
-target-versions = ["py38", "py39", "py310"]
+target-versions = ["py38", "py39", "py310", "py311"]
 
 [tool.poetry]
 name = "replay-rec"
@@ -40,29 +40,32 @@ classifiers = [
 exclude = [
     "replay/conftest.py",
 ]
-version = "0.17.1.preview"
+version = "0.18.0.preview"
 
 [tool.poetry.dependencies]
-python = ">=3.8.1, <3.11"
+python = ">=3.8.1, <3.12"
 numpy = ">=1.20.0"
-pandas = ">=1.3.5,<=2.2.2"
-polars = "~0.20.7"
+pandas = ">=1.3.5, <=2.2.2"
+polars = "~1.0.0"
 optuna = "~3.2.0"
-scipy = "~1.8.1"
-psutil = "~5.9.5"
-pyspark = {version = ">=3.0,<3.5", optional = true}
+scipy = "^1.8.1"
+psutil = "~6.0.0"
 scikit-learn = "^1.0.2"
 pyarrow = ">=12.0.1"
-nmslib = "2.1.1"
-hnswlib = "0.7.0"
-torch = "^1.8"
-lightning = "^2.0.2"
+fixed-install-nmslib = "2.1.2"
+hnswlib = "^0.7.0"
+pyspark = [
+    {version = ">=3.4,<3.5", python = ">=3.11,<3.12"},
+    {version = ">=3.0,<3.5", python = ">=3.8.1,<3.11"},
+]
+torch = ">=1.8, <=2.4.0"
+lightning = ">=2.0.2, <=2.4.0"
 pytorch-ranger = "^0.1.1"
 lightfm = "1.17"
 lightautoml = "~0.3.1"
 numba = ">=0.50"
 llvmlite = ">=0.32.1"
-sb-obp = "^0.5.7"
+sb-obp = "^0.5.8"
 d3rlpy = "^2.0.4"
 implicit = "~0.7.0"
 gym = "^0.26.0"
@@ -77,7 +80,7 @@ jupyter = "~1.0.0"
 jupyterlab = "^3.6.0"
 pytest = ">=7.1.0"
 pytest-cov = ">=3.0.0"
-statsmodels = "~0.13.5"
+statsmodels = "~0.14.0"
 black = ">=23.3.0"
 ruff = ">=0.0.261"
 toml-sort = "^0.23.0"
@@ -92,7 +95,7 @@ data-science-types = "0.2.23"
 
 [tool.poetry-dynamic-versioning]
 enable = false
-format-jinja = """0.17.1{{ env['PACKAGE_SUFFIX'] }}"""
+format-jinja = """0.18.0{{ env['PACKAGE_SUFFIX'] }}"""
 vcs = "git"
 
 [tool.ruff]

replay_rec-0.18.0rc0/replay/__init__.py

@@ -0,0 +1,3 @@
+""" RecSys library """
+
+__version__ = "0.18.0.preview"

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/data/dataset.py

@@ -1,6 +1,7 @@
 """
 ``Dataset`` universal dataset class for manipulating interactions and feed data to models.
 """
+
 from __future__ import annotations
 
 import json
@@ -606,7 +607,7 @@ class Dataset:
         if self.is_pandas:
             min_id = data[column].min()
         elif self.is_spark:
-            min_id = data.agg(sf.min(column).alias("min_index")).collect()[0][0]
+            min_id = data.agg(sf.min(column).alias("min_index")).first()[0]
         else:
             min_id = data[column].min()
         if min_id < 0:
@@ -616,7 +617,7 @@ class Dataset:
         if self.is_pandas:
             max_id = data[column].max()
         elif self.is_spark:
-            max_id = data.agg(sf.max(column).alias("max_index")).collect()[0][0]
+            max_id = data.agg(sf.max(column).alias("max_index")).first()[0]
         else:
             max_id = data[column].max()
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/data/dataset_utils/dataset_label_encoder.py

@@ -4,6 +4,7 @@ Contains classes for encoding categorical data
 ``LabelEncoderTransformWarning`` new category of warning for DatasetLabelEncoder.
 ``DatasetLabelEncoder`` to encode categorical features in `Dataset` objects.
 """
+
 import warnings
 from typing import Dict, Iterable, Iterator, Optional, Sequence, Set, Union
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/data/nn/schema.py

@@ -418,11 +418,11 @@ class TensorSchema(Mapping[str, TensorFeatureInfo]):
                 "feature_type": feature.feature_type.name,
                 "is_seq": feature.is_seq,
                 "feature_hint": feature.feature_hint.name if feature.feature_hint else None,
-                "feature_sources": [
-                    {"source": x.source.name, "column": x.column, "index": x.index} for x in feature.feature_sources
-                ]
-                if feature.feature_sources
-                else None,
+                "feature_sources": (
+                    [{"source": x.source.name, "column": x.column, "index": x.index} for x in feature.feature_sources]
+                    if feature.feature_sources
+                    else None
+                ),
                 "cardinality": feature.cardinality if feature.feature_type == FeatureType.CATEGORICAL else None,
                 "embedding_dim": feature.embedding_dim if feature.feature_type == FeatureType.CATEGORICAL else None,
                 "tensor_dim": feature.tensor_dim if feature.feature_type == FeatureType.NUMERICAL else None,

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/metrics/__init__.py

@@ -47,6 +47,7 @@ For each metric, a formula for its calculation is given, because this is
 important for the correct comparison of algorithms, as mentioned in our
 `article <https://arxiv.org/abs/2206.12858>`_.
 """
+
 from replay.experimental.metrics.base_metric import Metric, NCISMetric
 from replay.experimental.metrics.coverage import Coverage
 from replay.experimental.metrics.hitrate import HitRate

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/metrics/base_metric.py

@@ -1,6 +1,7 @@
 """
 Base classes for quality and diversity metrics.
 """
+
 import logging
 from abc import ABC, abstractmethod
 from typing import Dict, List, Optional, Union

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/models/base_rec.py

@@ -86,8 +86,8 @@ class BaseRecommender(RecommenderCommons, IsSavable, ABC):
         self.fit_items = sf.broadcast(items)
         self._num_users = self.fit_users.count()
         self._num_items = self.fit_items.count()
-        self._user_dim_size = self.fit_users.agg({"user_idx": "max"}).collect()[0][0] + 1
-        self._item_dim_size = self.fit_items.agg({"item_idx": "max"}).collect()[0][0] + 1
+        self._user_dim_size = self.fit_users.agg({"user_idx": "max"}).first()[0] + 1
+        self._item_dim_size = self.fit_items.agg({"item_idx": "max"}).first()[0] + 1
         self._fit(log, user_features, item_features)
 
     @abstractmethod
@@ -122,7 +122,7 @@ class BaseRecommender(RecommenderCommons, IsSavable, ABC):
         # count maximal number of items seen by users
         max_seen = 0
         if num_seen.count() > 0:
-            max_seen = num_seen.select(sf.max("seen_count")).collect()[0][0]
+            max_seen = num_seen.select(sf.max("seen_count")).first()[0]
 
         # crop recommendations to first k + max_seen items for each user
         recs = recs.withColumn(
@@ -335,7 +335,7 @@ class BaseRecommender(RecommenderCommons, IsSavable, ABC):
             setattr(
                 self,
                 f"_{entity}_dim_size",
-                getattr(self, f"fit_{entity}s").agg({f"{entity}_idx": "max"}).collect()[0][0] + 1,
+                getattr(self, f"fit_{entity}s").agg({f"{entity}_idx": "max"}).first()[0] + 1,
             )
         return getattr(self, f"_{entity}_dim_size")
 
@@ -1088,7 +1088,7 @@ class NonPersonalizedRecommender(Recommender, ABC):
         Calculating a fill value a the minimal relevance
         calculated during model training multiplied by weight.
         """
-        return item_popularity.select(sf.min("relevance")).collect()[0][0] * weight
+        return item_popularity.select(sf.min("relevance")).first()[0] * weight
 
     @staticmethod
     def _check_relevance(log: SparkDataFrame):
@@ -1113,7 +1113,7 @@ class NonPersonalizedRecommender(Recommender, ABC):
         max_hist_len = (
             (log.join(users, on="user_idx").groupBy("user_idx").agg(sf.countDistinct("item_idx").alias("items_count")))
             .select(sf.max("items_count"))
-            .collect()[0][0]
+            .first()[0]
         )
         # all users have empty history
         if max_hist_len is None:
@@ -1146,7 +1146,7 @@ class NonPersonalizedRecommender(Recommender, ABC):
             users = users.join(user_to_num_items, on="user_idx", how="left")
             users = users.fillna(0, "num_items")
             # 'selected_item_popularity' truncation by k + max_seen
-            max_seen = users.select(sf.coalesce(sf.max("num_items"), sf.lit(0))).collect()[0][0]
+            max_seen = users.select(sf.coalesce(sf.max("num_items"), sf.lit(0))).first()[0]
             selected_item_popularity = selected_item_popularity.filter(sf.col("rank") <= k + max_seen)
             return users.join(selected_item_popularity, on=(sf.col("rank") <= k + sf.col("num_items")), how="left")
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/models/cql.py

@@ -1,6 +1,7 @@
 """
 Using CQL implementation from `d3rlpy` package.
 """
+
 import io
 import logging
 import tempfile
@@ -402,6 +403,7 @@ class MdpDatasetBuilder:
         top_k (int): the number of top user items to learn predicting.
         action_randomization_scale (float): the scale of action randomization gaussian noise.
     """
+
     logger: logging.Logger
     top_k: int
     action_randomization_scale: float

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/models/ddpg.py

@@ -704,13 +704,15 @@ class DDPG(Recommender):
         :param data: pandas DataFrame
         """
         data = data[["user_idx", "item_idx", "relevance"]]
-        train_data = data.values.tolist()
+        users = data["user_idx"].values.tolist()
+        items = data["item_idx"].values.tolist()
+        scores = data["relevance"].values.tolist()
 
-        user_num = data["user_idx"].max() + 1
-        item_num = data["item_idx"].max() + 1
+        user_num = max(users) + 1
+        item_num = max(items) + 1
 
         train_mat = defaultdict(float)
-        for user, item, rel in train_data:
+        for user, item, rel in zip(users, items, scores):
             train_mat[user, item] = rel
         train_matrix = sp.dok_matrix((user_num, item_num), dtype=np.float32)
         dict.update(train_matrix, train_mat)

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/models/lightfm_wrap.py

@@ -98,12 +98,12 @@ class LightFMWrap(HybridRecommender):
         fit_dim = getattr(self, f"_{entity}_dim")
         matrix_height = max(
             fit_dim,
-            log_ids_list.select(sf.max(idx_col_name)).collect()[0][0] + 1,
+            log_ids_list.select(sf.max(idx_col_name)).first()[0] + 1,
         )
         if not feature_table.rdd.isEmpty():
             matrix_height = max(
                 matrix_height,
-                feature_table.select(sf.max(idx_col_name)).collect()[0][0] + 1,
+                feature_table.select(sf.max(idx_col_name)).first()[0] + 1,
             )
 
         features_np = (

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/models/mult_vae.py

@@ -2,6 +2,7 @@
 MultVAE implementation
 (Variational Autoencoders for Collaborative Filtering)
 """
+
 from typing import Optional, Tuple
 
 import numpy as np

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/models/neuromf.py

@@ -3,6 +3,7 @@ Generalized Matrix Factorization (GMF),
 Multi-Layer Perceptron (MLP),
 Neural Matrix Factorization (MLP + GMF).
 """
+
 from typing import List, Optional
 
 import numpy as np

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/models/scala_als.py

@@ -115,7 +115,7 @@ class ALSWrap(Recommender, ItemVectorModel):
                     .groupBy("user_idx")
                     .agg(sf.count("user_idx").alias("num_seen"))
                     .select(sf.max("num_seen"))
-                    .collect()[0][0]
+                    .first()[0]
                 )
                 max_seen = max_seen_in_log if max_seen_in_log is not None else 0
 
@@ -280,7 +280,7 @@ class ScalaALSWrap(ALSWrap, ANNMixin):
                 .groupBy("user_idx")
                 .agg(sf.count("user_idx").alias("num_seen"))
                 .select(sf.max("num_seen"))
-                .collect()[0][0]
+                .first()[0]
             )
             max_seen = max_seen_in_log if max_seen_in_log is not None else 0
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/preprocessing/data_preparator.py

@@ -6,6 +6,7 @@ Contains classes for data preparation and categorical features transformation.
 ``ToNumericFeatureTransformer`` leaves only numerical features
 by one-hot encoding of some features and deleting the others.
 """
+
 import json
 import logging
 import string
@@ -699,7 +700,7 @@ if PYSPARK_AVAILABLE:
                 return
 
             cat_feat_values_dict = {
-                name: (spark_df.select(sf.collect_set(sf.col(name))).collect()[0][0]) for name in self.cat_cols_list
+                name: (spark_df.select(sf.collect_set(sf.col(name))).first()[0]) for name in self.cat_cols_list
             }
             self.expressions_list = [
                 sf.when(sf.col(col_name) == cur_name, 1)

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/preprocessing/padder.py

@@ -179,7 +179,7 @@ class Padder:
         self, df_transformed: SparkDataFrame, col: str, pad_value: Union[str, float, List, None]
     ) -> SparkDataFrame:
         if self.array_size == -1:
-            max_array_size = df_transformed.agg(sf.max(sf.size(col)).alias("max_array_len")).collect()[0][0]
+            max_array_size = df_transformed.agg(sf.max(sf.size(col)).alias("max_array_len")).first()[0]
         else:
             max_array_size = self.array_size
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/scenarios/two_stages/two_stages_scenario.py

@@ -383,7 +383,7 @@ class TwoStagesScenario(HybridRecommender):
                 log_to_filter_cached.groupBy("user_idx")
                 .agg(sf.count("item_idx").alias("num_positives"))
                 .select(sf.max("num_positives"))
-                .collect()[0][0]
+                .first()[0]
             )
 
         pred = model._predict(

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/experimental/utils/model_handler.py

@@ -170,8 +170,13 @@ def load_indexer(path: str) -> Indexer:
 
     indexer = Indexer(**args)
 
-    indexer.user_type = getattr(st, user_type)()
-    indexer.item_type = getattr(st, item_type)()
+    if user_type.endswith("()"):
+        user_type = user_type[:-2]
+        item_type = item_type[:-2]
+    user_type = getattr(st, user_type)
+    item_type = getattr(st, item_type)
+    indexer.user_type = user_type()
+    indexer.item_type = item_type()
 
     indexer.user_indexer = StringIndexerModel.load(join(path, "user_indexer"))
     indexer.item_indexer = StringIndexerModel.load(join(path, "item_indexer"))

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/metrics/__init__.py

@@ -42,6 +42,7 @@ For each metric, a formula for its calculation is given, because this is
 important for the correct comparison of algorithms, as mentioned in our
 `article <https://arxiv.org/abs/2206.12858>`_.
 """
+
 from .base_metric import Metric
 from .categorical_diversity import CategoricalDiversity
 from .coverage import Coverage

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/models/als.py

@@ -115,7 +115,7 @@ class ALSWrap(Recommender, ItemVectorModel):
                     .groupBy(self.query_column)
                     .agg(sf.count(self.query_column).alias("num_seen"))
                     .select(sf.max("num_seen"))
-                    .collect()[0][0]
+                    .first()[0]
                 )
                 max_seen = max_seen_in_interactions if max_seen_in_interactions is not None else 0
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/models/base_rec.py

@@ -401,8 +401,8 @@ class BaseRecommender(RecommenderCommons, IsSavable, ABC):
         self.fit_items = sf.broadcast(items)
         self._num_queries = self.fit_queries.count()
         self._num_items = self.fit_items.count()
-        self._query_dim_size = self.fit_queries.agg({self.query_column: "max"}).collect()[0][0] + 1
-        self._item_dim_size = self.fit_items.agg({self.item_column: "max"}).collect()[0][0] + 1
+        self._query_dim_size = self.fit_queries.agg({self.query_column: "max"}).first()[0] + 1
+        self._item_dim_size = self.fit_items.agg({self.item_column: "max"}).first()[0] + 1
         self._fit(dataset)
 
     @abstractmethod
@@ -431,7 +431,7 @@ class BaseRecommender(RecommenderCommons, IsSavable, ABC):
         # count maximal number of items seen by queries
         max_seen = 0
         if num_seen.count() > 0:
-            max_seen = num_seen.select(sf.max("seen_count")).collect()[0][0]
+            max_seen = num_seen.select(sf.max("seen_count")).first()[0]
 
         # crop recommendations to first k + max_seen items for each query
         recs = recs.withColumn(
@@ -708,7 +708,7 @@ class BaseRecommender(RecommenderCommons, IsSavable, ABC):
             setattr(
                 self,
                 dim_size,
-                fit_entities.agg({column: "max"}).collect()[0][0] + 1,
+                fit_entities.agg({column: "max"}).first()[0] + 1,
             )
         return getattr(self, dim_size)
 
@@ -1426,7 +1426,7 @@ class NonPersonalizedRecommender(Recommender, ABC):
         Calculating a fill value a the minimal rating
         calculated during model training multiplied by weight.
         """
-        return item_popularity.select(sf.min(rating_column)).collect()[0][0] * weight
+        return item_popularity.select(sf.min(rating_column)).first()[0] * weight
 
     @staticmethod
     def _check_rating(dataset: Dataset):
@@ -1460,7 +1460,7 @@ class NonPersonalizedRecommender(Recommender, ABC):
                 .agg(sf.countDistinct(item_column).alias("items_count"))
             )
             .select(sf.max("items_count"))
-            .collect()[0][0]
+            .first()[0]
         )
         # all queries have empty history
         if max_hist_len is None:
@@ -1495,7 +1495,7 @@ class NonPersonalizedRecommender(Recommender, ABC):
             queries = queries.join(query_to_num_items, on=self.query_column, how="left")
             queries = queries.fillna(0, "num_items")
             # 'selected_item_popularity' truncation by k + max_seen
-            max_seen = queries.select(sf.coalesce(sf.max("num_items"), sf.lit(0))).collect()[0][0]
+            max_seen = queries.select(sf.coalesce(sf.max("num_items"), sf.lit(0))).first()[0]
             selected_item_popularity = selected_item_popularity.filter(sf.col("rank") <= k + max_seen)
             return queries.join(selected_item_popularity, on=(sf.col("rank") <= k + sf.col("num_items")), how="left")
 

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/models/extensions/ann/index_inferers/nmslib_filter_index_inferer.py

@@ -32,9 +32,9 @@ class NmslibFilterIndexInferer(IndexInferer):
             index = index_store.load_index(
                 init_index=lambda: create_nmslib_index_instance(index_params),
                 load_index=lambda index, path: index.loadIndex(path, load_data=True),
-                configure_index=lambda index: index.setQueryTimeParams({"efSearch": index_params.ef_s})
-                if index_params.ef_s
-                else None,
+                configure_index=lambda index: (
+                    index.setQueryTimeParams({"efSearch": index_params.ef_s}) if index_params.ef_s else None
+                ),
             )
 
             # max number of items to retrieve per batch

{replay_rec-0.17.1rc0 → replay_rec-0.18.0rc0}/replay/models/extensions/ann/index_inferers/nmslib_index_inferer.py

@@ -30,9 +30,9 @@ class NmslibIndexInferer(IndexInferer):
             index = index_store.load_index(
                 init_index=lambda: create_nmslib_index_instance(index_params),
                 load_index=lambda index, path: index.loadIndex(path, load_data=True),
-                configure_index=lambda index: index.setQueryTimeParams({"efSearch": index_params.ef_s})
-                if index_params.ef_s
-                else None,
+                configure_index=lambda index: (
+                    index.setQueryTimeParams({"efSearch": index_params.ef_s}) if index_params.ef_s else None
+                ),
             )
 
             user_vectors = get_csr_matrix(user_idx, vector_items, vector_ratings)