Fix negative sampler config (#278)

Fixed negative sampler kwargs in transformer model config

Author: spirinamayya

rectools/models/nn/transformers/base.py

@@ -225,6 +225,7 @@ class TransformerModelConfig(ModelConfig):
     item_net_constructor_kwargs: tp.Optional[InitKwargs] = None
     pos_encoding_kwargs: tp.Optional[InitKwargs] = None
     lightning_module_kwargs: tp.Optional[InitKwargs] = None
+    negative_sampler_kwargs: tp.Optional[InitKwargs] = None
     similarity_module_kwargs: tp.Optional[InitKwargs] = None
     backbone_kwargs: tp.Optional[InitKwargs] = None
 

tests/models/nn/transformers/test_bert4rec.py

@@ -865,6 +865,7 @@ def initial_config(self) -> tp.Dict[str, tp.Any]:
             "item_net_constructor_kwargs": None,
             "pos_encoding_kwargs": None,
             "lightning_module_kwargs": None,
+            "negative_sampler_kwargs": None,
             "similarity_module_kwargs": None,
             "backbone_kwargs": None,
         }

tests/models/nn/transformers/test_sasrec.py

@@ -973,6 +973,7 @@ def initial_config(self) -> tp.Dict[str, tp.Any]:
             "item_net_constructor_kwargs": None,
             "pos_encoding_kwargs": None,
             "lightning_module_kwargs": None,
+            "negative_sampler_kwargs": None,
             "similarity_module_kwargs": None,
             "backbone_kwargs": None,
         }