diff options
Diffstat (limited to 'ot/gpu/__init__.py')
| -rw-r--r-- | ot/gpu/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py index 9319208..40b11c0 100644 --- a/ot/gpu/__init__.py +++ b/ot/gpu/__init__.py @@ -2,5 +2,6 @@ from . import bregman from . import da +from .bregman import sinkhorn -__all__ = ["bregman", "da"] +__all__ = ["bregman", "da", "sinkhorn"] |