Updates per style check

rlratzel · Aug 12, 2020 · 3d30eb0 · 3d30eb0
1 parent 38408b9
commit 3d30eb0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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
+    assert leiden_mod >= louvain_mod