fix style and add comment

rapidsai · Nov 22, 2023 · 9f8b131 · 9f8b131
1 parent 71fb5e5
commit 9f8b131
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/cugraph/cugraph/tests/conftest.py b/python/cugraph/cugraph/tests/conftest.py
@@ -19,6 +19,7 @@
 
 import os
 import tempfile
+# Avoid timeout during shutdown
 from dask_cuda.utils_test import IncreasedCloseTimeoutNanny
 
 # module-wide fixtures
@@ -42,7 +43,8 @@ def dask_client():
     # DASK_WORKER_DEVICES env vars and use them when creating a client if
     # set. start_dask_client will also initialize the Comms singleton.
     dask_client, dask_cluster = start_dask_client(
-        worker_class=IncreasedCloseTimeoutNanny)
+        worker_class=IncreasedCloseTimeoutNanny
+    )
 
     yield dask_client
 

diff --git a/python/cugraph/cugraph/tests/link_analysis/test_hits_mg.py b/python/cugraph/cugraph/tests/link_analysis/test_hits_mg.py
@@ -45,7 +45,7 @@ def setup_function():
 fixture_params = gen_fixture_params_product(
     (datasets, "graph_file"),
     ([50], "max_iter"),
-    ([1.0e-4], "tol"), # FIXME: Temporarily lower tolerance
+    ([1.0e-4], "tol"),  # FIXME: Temporarily lower tolerance
     (IS_DIRECTED, "directed"),
 )