diff options
Diffstat (limited to 'ot/gpu/__init__.py')
-rw-r--r-- | ot/gpu/__init__.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py index a2fdd3d..ed6dcc4 100644 --- a/ot/gpu/__init__.py +++ b/ot/gpu/__init__.py @@ -9,4 +9,9 @@ from .bregman import sinkhorn # # License: MIT License +import warnings + +warnings.warn("the ot.gpu module is deprecated because cudamat in no longer maintained", DeprecationWarning, + stacklevel=2) + __all__ = ["bregman", "da", "sinkhorn"] |