test fixes

This commit is contained in:
Liam Brydon
2025-11-07 08:46:28 +13:00
parent dddea79daf
commit 9ec5e433ea

View File

@ -2853,7 +2853,6 @@ class ApplicabilityDomain(EnviPathModel):
train_instances = train_instances.with_columns(dist=pl.Series(dists))
# sort them in a descending way and take at most `self.num_neighbours`
# TODO: Should this be descending? If we want the most similar then we want values close to zero (ascending)
train_instances = train_instances.sort("dist", descending=True)[:self.num_neighbours]
# compute average distance
rule_reliabilities[rule_uuid] = train_instances.select(pl.mean("dist")).fill_nan(0.0).item()