Networkx 2.0 compatibility fix by @Erotemic (#75)

gtaylor · Aug 21, 2017 · 30bd9cb · 30bd9cb
1 parent aaa425a
commit 30bd9cb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/colormath/color_conversions.py b/colormath/color_conversions.py
@@ -122,7 +122,10 @@ def get_conversion_path(self, start_type, target_type):
 
     def add_type_conversion(self, start_type, target_type, conversion_function):
         super(GraphConversionManager, self).add_type_conversion(start_type, target_type, conversion_function)
-        self.conversion_graph.add_edge(start_type, target_type, {'conversion_function': conversion_function})
+        if networkx.__version__.startswith('2'):
+            self.conversion_graph.add_edge(start_type, target_type, conversion_function=conversion_function)
+        else:
+            self.conversion_graph.add_edge(start_type, target_type, {'conversion_function': conversion_function})
 
 
 class DummyConversionManager(ConversionManager):