Skip to content

Commit a1b8af7

Browse files
committed
Do not explicitly set device in curve cosntructor as it follows from the other input arguments
1 parent 9e11afe commit a1b8af7

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

stochman/curves.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ def __init__(
1313
end: torch.Tensor,
1414
num_nodes: int = 5,
1515
requires_grad: bool = True,
16+
device = None,
1617
*args,
1718
**kwargs,
1819
) -> None:

0 commit comments

Comments
 (0)