Skip to content

Commit fa7f3dd

Browse files
committed
correction import in ot.gpu
1 parent ee8ed4f commit fa7f3dd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ot/gpu/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
# -*- coding: utf-8 -*-
22
"""
33
4-
5-
This module implement GPU ilmplementation for several OT solvers and utility
4+
This module provides GPU implementation for several OT solvers and utility
65
functions. The GPU backend in handled by `cupy
76
<https://cupy.chainer.org/>`_.
87
@@ -22,7 +21,7 @@
2221
from . import bregman
2322
from . import da
2423
from .bregman import sinkhorn
25-
from .da
24+
from .da import sinkhorn_lpl1_mm
2625

2726
from . import utils
2827
from .utils import dist, to_gpu, to_np
@@ -33,4 +32,5 @@
3332
#
3433
# License: MIT License
3534

36-
__all__ = ["utils", "dist", "sinkhorn", 'bregman', 'da', 'to_gpu', 'to_np']
35+
__all__ = ["utils", "dist", "sinkhorn",
36+
"sinkhorn_lpl1_mm", 'bregman', 'da', 'to_gpu', 'to_np']

0 commit comments

Comments
 (0)