Skip to content

Commit 81ea20c

Browse files
committed
Seed UMAP kwargs in DBSCAN regression test
1 parent 2bd4767 commit 81ea20c

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

graphistry/tests/test_compute_cluster.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,9 @@ def _condition(self, g: Plottable, kind):
7272
def test_umap_cluster(self):
7373
g = graphistry.nodes(ndf).edges(edf, 'src', 'dst')
7474
for kind in ['nodes', 'edges']:
75-
g2 = g.umap(kind=kind, n_topics=2, dbscan=False, random_state=42).dbscan(kind=kind, verbose=True)
75+
g2 = g.umap(kind=kind, n_topics=2, dbscan=False, umap_kwargs={"random_state": 42}).dbscan(kind=kind, verbose=True)
7676
self._condition(g2, kind)
77-
g3 = g.umap(kind=kind, n_topics=2, dbscan=True, random_state=42)
77+
g3 = g.umap(kind=kind, n_topics=2, dbscan=True, umap_kwargs={"random_state": 42})
7878
self._condition(g3, kind)
7979
if kind == 'nodes':
8080
self.assertTrue(_labels_equivalent_up_to_permutation(g2._nodes[DBSCAN], g3._nodes[DBSCAN]))

0 commit comments

Comments
 (0)