up todo

Author: Daria Tikhonovich

rectools/models/nn/transformers/context_net.py

@@ -77,5 +77,5 @@ def forward(self, seqs: torch.Tensor, batch: tp.Dict[str, torch.Tensor]) -> torc
 
     @property
     def out_dim(self) -> int:
-        """Return categorical item embedding output dimension."""
+        """Return output dimension."""
         return self.embedding_bag.embedding_dim