Merge pull request #5 from belowlab/refactor

feat: switched from using typer to just argparse
belowlab · Nov 27, 2023 · 192dbb4 · 192dbb4
2 parents e212f57 + f6c3d6c
commit 192dbb4
Show file tree

Hide file tree

Showing 10 changed files with 355 additions and 734 deletions.
diff --git a/drive/cluster/cluster.py b/drive/cluster/cluster.py
@@ -77,6 +77,9 @@ def random_walk(self, graph: ig.Graph) -> ig.VertexClustering:
             result of the random walk cluster. This object has
             information about clusters and membership
         """
+        logger.debug(
+            f"Performing the community walktrap algorithm with a random walk step size of: {self.random_walk_step_size}"
+        )
 
         ibd_walktrap = ig.Graph.community_walktrap(
             graph, weights="cm", steps=self.random_walk_step_size