Skip to content

Commit 825c00b

Browse files
committed
TEST: Fit from neighbours with fixed self-counting
1 parent 464961b commit 825c00b

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

cnnclustering/cnn.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2333,7 +2333,8 @@ def predict(
23332333
other.labels.consider,
23342334
self.labels,
23352335
set(clusters),
2336-
_cnn_cutoff)
2336+
params["cnn_cutoff"],
2337+
other.data.neighbourhoods.self_counting)
23372338

23382339
# Predict from List[Set[int]]
23392340
# TODO: Allow different methods and data structures

tests/unit/test_clustering.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def test_fit_from_NeighbourhoodsList_base_neighbourhoods_e15(
7878
labels = np.zeros(len(neighbourhoods), dtype=np.int_)
7979
consider = np.ones_like(labels, dtype=np.uint8)
8080
cfits.fit_from_NeighbourhoodsList(
81-
neighbourhoods, labels, consider, c
81+
neighbourhoods, labels, consider, c, False
8282
)
8383

8484
np.testing.assert_array_equal(
@@ -101,7 +101,7 @@ def test_fit_from_NeighbourhoodsArray_base_neighbourhoods_e15(
101101
labels = np.zeros(len(neighbourhoods), dtype=np.int_)
102102
consider = np.ones_like(labels, dtype=np.uint8)
103103
cfits.fit_from_NeighbourhoodsArray(
104-
neighbourhoods, labels, consider, c
104+
neighbourhoods, labels, consider, c, False
105105
)
106106

107107
np.testing.assert_array_equal(

0 commit comments

Comments
 (0)