Skip to content

Commit a70cf8d

Browse files
author
Francisco Muñoz
committed
refactor: run pre-commit
1 parent 1030815 commit a70cf8d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ot/bregman/_convolutional.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def _get_convol_img_fn(nx, width, height, reg, type_as, log_domain=False):
3131
t2 = nx.linspace(0, 1, height, type_as=type_as)
3232
Y2, X2 = nx.meshgrid(t2, t2)
3333
M2 = -((X2 - Y2) ** 2) / reg
34-
34+
3535
# If normal domain is selected, we can use M1 and M2 to compute the convolution
3636
if not log_domain:
3737
K1, K2 = nx.exp(M1), nx.exp(M2)
@@ -43,6 +43,7 @@ def convol_imgs(imgs):
4343

4444
# Else, we can use M1 and M2 to compute the convolution in log-domain
4545
else:
46+
4647
def convol_imgs(log_imgs):
4748
log_imgs = nx.logsumexp(M1[:, :, None] + log_imgs[None], axis=1)
4849
log_imgs = nx.logsumexp(M2[:, :, None] + log_imgs.T[None], axis=1).T

0 commit comments

Comments
 (0)