adding an n_neighbors arg

brifordwylie · brifordwylie · commit e941affd4004 · 2025-01-11T15:01:50.000-07:00
diff --git a/src/workbench/algorithms/graph/light/proximity_graph.py b/src/workbench/algorithms/graph/light/proximity_graph.py
@@ -183,7 +183,7 @@ def get_neighborhood(self, node_id: Union[str, int], radius: int = 1) -> nx.Grap
         df,
         id_column=fs.id_column,
         proximity_class=FingerprintProximity,
-        proximity_kwargs={"fingerprint_column": "morgan_fingerprint"},
+        proximity_kwargs={"fingerprint_column": "morgan_fingerprint", "n_neighbors": 2},
     )
     nx_graph = fingerprint_graph.get_graph()
 

Original file line number	Diff line number	Diff line change
`@@ -183,7 +183,7 @@ def get_neighborhood(self, node_id: Union[str, int], radius: int = 1) -> nx.Grap`
`183`	`183`	`df,`
`184`	`184`	`id_column=fs.id_column,`
`185`	`185`	`proximity_class=FingerprintProximity,`
`186`		`- proximity_kwargs={"fingerprint_column": "morgan_fingerprint"},`
	`186`	`+ proximity_kwargs={"fingerprint_column": "morgan_fingerprint", "n_neighbors": 2},`
`187`	`187`	`)`
`188`	`188`	`nx_graph = fingerprint_graph.get_graph()`
`189`	`189`