Skip to content

Commit 9762b75

Browse files
committed
fix np array np.int64 conversion for python3
1 parent fb7b9db commit 9762b75

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/VertexPartition.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from . import _c_louvain
33
from .functions import _get_py_capsule
44
import sys
5+
import numpy as np
56
# Check if working with Python 3
67
PY3 = (sys.version > '3')
78

@@ -849,8 +850,15 @@ def __init__(self, graph, layer_vec=None, initial_membership=None, weights=None,
849850
else:
850851
# Make sure it is a list
851852
weights = list(weights)
853+
854+
852855
if layer_vec is not None:
853-
layer_vec = list(layer_vec) # ensure that it is a list
856+
layer_vec=np.array(layer_vec).tolist()
857+
# if isinstance(layer_vec, np.ndarray):
858+
# layer_vec = layer_vec.tolist()
859+
# elif layer_vec is not None:
860+
# layer_vec = list(layer_vec)
861+
854862
else:
855863
layer_vec = [0 for _ in range(graph.vcount())] #default is to assume single layer
856864

0 commit comments

Comments
 (0)