diff --git a/netbone/filters.py b/netbone/filters.py index 2dacd360ac72dfa02b3b53649708b990d63b12d3..a728f80bb5673e4f5030812af9be51d2cd3da030 100644 --- a/netbone/filters.py +++ b/netbone/filters.py @@ -60,16 +60,9 @@ def fraction_filter(backbone, value, narrate=True, secondary_property='weight', value = math.ceil(value * len(data)) return nx.from_pandas_edgelist(data[:value], edge_attr=edge_properties(data)) else: - print('hereeeeeeee') - b = backbone.graph.copy() value = math.ceil(value * len(backbone.graph)) - - print(value) - - print(list(data[:value].index)) - b.remove_nodes_from(list(data[:value].index)) - return b + return b.subgraph(list(data[:value].index)) print("The accepted filters for " + backbone.method_name + " are: " + ', '.join( [fun.__name__ for fun in backbone.compatible_filters()]))