From 3d30eb0e0aa4d97bcf50a9503ee258865a7c7f7b Mon Sep 17 00:00:00 2001 From: James Wyles Date: Wed, 12 Aug 2020 17:05:13 -0600 Subject: [PATCH] Updates per style check --- python/cugraph/tests/test_leiden.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/cugraph/tests/test_leiden.py b/python/cugraph/tests/test_leiden.py index 157271c80ba..b6dfdc7dd70 100644 --- a/python/cugraph/tests/test_leiden.py +++ b/python/cugraph/tests/test_leiden.py @@ -28,8 +28,6 @@ with warnings.catch_warnings(): warnings.filterwarnings("ignore", category=DeprecationWarning) - import community - import networkx as nx print("Networkx version : {} ".format(nx.__version__)) @@ -50,6 +48,7 @@ def cugraph_leiden(cu_M, edgevals=False): return parts, mod + def cugraph_louvain(cu_M, edgevals=False): G = cugraph.Graph() @@ -65,6 +64,7 @@ def cugraph_louvain(cu_M, edgevals=False): return parts, mod + @pytest.mark.parametrize("graph_file", utils.UNRENUMBERED_DATASETS) def test_louvain(graph_file): gc.collect() @@ -74,4 +74,4 @@ def test_louvain(graph_file): louvain_parts, louvain_mod = cugraph_louvain(cu_M, edgevals=True) # Calculating modularity scores for comparison - assert leiden_mod >= louvain_mod \ No newline at end of file + assert leiden_mod >= louvain_mod