Skip to content

Commit 78b57d6

Browse files
committed
Explicit use of apply to fix deprecation warning
1 parent 0054996 commit 78b57d6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stochman/utilities/distance.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,4 +46,4 @@ def squared_manifold_distance(manifold, p0: torch.Tensor, p1: torch.Tensor):
4646
dist: squared distance from p0 to p1 calculated on the manifold
4747
"""
4848
distance_op = __Dist2__()
49-
return distance_op(manifold, p0, p1)
49+
return distance_op.apply(manifold, p0, p1)

0 commit comments

Comments
 (0)