1 files changed, 4 insertions, 5 deletions
diff --git a/ot/__init__.py b/ot/__init__.py
index 3cf8e0b..24350a5 100644
--- a/ot/__init__.py
+++ b/ot/__init__.py
@@ -1,18 +1,17 @@
-
 # Python Optimal Transport toolbox
 
-
+# All submodules and packages
 import utils
 import datasets
 import plot
+import bregman
 
 
-# Ot functions
+# OT functions
 from emd import emd
 from bregman import sinkhorn
 
-
-
+# utils functions
 from utils import dist,dots
 
 __all__ = ["emd","sinkhorn","utils",'datasets','plot','dist','dots']