Skip to content

Commit bc2ce4f

Browse files
authored
Add default dim to cat (similar to torch.cat) (#521)
1 parent fd14893 commit bc2ce4f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch_frame/utils/concat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131

3232
def cat(
3333
lst: list[T],
34-
dim: int,
34+
dim: int = 0,
3535
) -> T:
3636
r"""Concatenates a list of :class:`TensorFrame` or :class:`TensorData`.
3737

0 commit comments

Comments
 (0)