From 9d4b786a036ac95989825beec819521089fb4feb Mon Sep 17 00:00:00 2001
From: ngayraud <nat.gayraud@gmail.com>
Date: Mon, 12 Aug 2019 16:37:58 -0400
Subject: fixes for travis, added test, minor nits

---
 ot/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'ot/utils.py')

diff --git a/ot/utils.py b/ot/utils.py
index be839f8..a334fea 100644
--- a/ot/utils.py
+++ b/ot/utils.py
@@ -178,7 +178,9 @@ def cost_normalization(C, norm=None):
         The input cost matrix normalized according to given norm.
     """
 
-    if norm == "median":
+    if norm is None:
+        pass
+    elif norm == "median":
         C /= float(np.median(C))
     elif norm == "max":
         C /= float(np.max(C))
-- 
cgit v1.2.3