summaryrefslogtreecommitdiff
path: root/ot/__init__.py
diff options
context:
space:
mode:
authorLeo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr>2017-04-20 13:51:30 +0200
committerLeo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr>2017-04-20 13:51:30 +0200
commit67f09ad4089594457cb00702dd3aa8a6a94ec2eb (patch)
treed458e5661f8ae9dd1dfe8127003ac4141f74b7d0 /ot/__init__.py
parent16f51f971607efab2c73958d207c582b389406c8 (diff)
changes from feedback
Diffstat (limited to 'ot/__init__.py')
-rw-r--r--ot/__init__.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/ot/__init__.py b/ot/__init__.py
index 702269c..13abcda 100644
--- a/ot/__init__.py
+++ b/ot/__init__.py
@@ -9,7 +9,6 @@ from . import utils
from . import datasets
from . import plot
from . import da
-from . import gpu
# OT functions
from .lp import emd, emd2
@@ -23,5 +22,4 @@ __version__ = "0.2"
__all__ = ["emd", "emd2", "sinkhorn", "utils", 'datasets', 'bregman', 'lp',
'plot', 'tic', 'toc', 'toq',
- 'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim',
- 'gpu']
+ 'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim']