py2 compatibility

MaineKuehn · Nov 16, 2017 · e95e4ee · e95e4ee
1 parent 78d0438
commit e95e4ee
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/graphi/graph_io/csv.py b/graphi/graph_io/csv.py
@@ -1,6 +1,7 @@
 """
 Utilities for loading and storing Graphs
 """
+from __future__ import absolute_import
 import csv
 import ast
 import itertools

diff --git a/graphi/types/adjacency_graph.py b/graphi/types/adjacency_graph.py
@@ -1,5 +1,5 @@
 from __future__ import absolute_import
-from collections import abc as abc_collection
+import collections as abc_collection
 import six
 
 from .. import abc
@@ -22,11 +22,11 @@ class AdjacencyGraph(abc.Graph):
     O(len(:py:meth:`nodes`) = n) and O(len(:py:meth:`edges`) -> n\ :sup:`2`\ ),
     respectively.
     """
-    def __init__(self, *source, undirected=False):
-        self.undirected = undirected
+    def __init__(self, *source, **kwargs):
+        self.undirected = kwargs.pop('undirected', False)
         self._adjacency = {}  # {node: {neighbour: distance, neighbour: distance, ...}, ...}
         super(AdjacencyGraph, self).__init__(*source)
-        if undirected:
+        if self.undirected:
             self._ensure_symmetry()
 
     @staticmethod