Skip to content

Commit a2824fb

Browse files
committed
Add sorting to doc-string as pointed out in PR review
1 parent b2d1226 commit a2824fb

File tree

1 file changed

+4
-0
lines changed
  • merlin/models/torch/transforms

1 file changed

+4
-0
lines changed

merlin/models/torch/transforms/agg.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ def forward(self, inputs: Dict[str, torch.Tensor]) -> torch.Tensor:
3737
"""
3838
Concatenates input tensors along the specified dimension.
3939
40+
The input dictionary will be sorted by name before concatenation.
41+
4042
Parameters
4143
----------
4244
inputs : Dict[str, torch.Tensor]
@@ -77,6 +79,8 @@ def forward(self, inputs: Dict[str, torch.Tensor]) -> torch.Tensor:
7779
class Stack(nn.Module):
7880
"""Stack tensors along a specified dimension.
7981
82+
The input dictionary will be sorted by name before concatenation.
83+
8084
Parameters
8185
----------
8286
dim : int

0 commit comments

Comments
 (0)