Update __init__.py

ott-jax · May 21, 2022 · 991515a · 991515a
1 parent 0a78de0
commit 991515a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ott/tools/__init__.py b/ott/tools/__init__.py
@@ -15,7 +15,7 @@
 
 """OTT tools: A set of tools to use OT in differentiable ML pipelines."""
 
-#from . import plot
+from . import plot
 from . import gaussian_mixture
 from . import sinkhorn_divergence
 from . import soft_sort