Updating HSS

netbone/structural/high_salience_skeleton.py

@@ -8,9 +8,10 @@ from netbone.filters import boolean_filter, threshold_filter, fraction_filter
 
 # change distance
 def high_salience_skeleton(data):
-    graph = data.copy()
     if isinstance(data, pd.DataFrame):
-        graph = nx.from_pandas_edgelist(data)
+        graph = nx.from_pandas_edgelist(data, edge_attr='weight', create_using=nx.Graph())
+    elif isinstance(data, nx.Graph):
+        graph = data.copy()
     else:
         print("data should be a panads dataframe or nx graph")
         return