diff --git a/src/sensai/nearest_neighbors.py b/src/sensai/nearest_neighbors.py index af71c570..12f5238e 100644 --- a/src/sensai/nearest_neighbors.py +++ b/src/sensai/nearest_neighbors.py @@ -299,7 +299,7 @@ def __init__(self, num_neighbors: int, distance_metric: DistanceMetric, self.knn_finder = None def _tostring_excludes(self) -> List[str]: - return super()._tostring_excludes() + ["neighborProviderFactory", "distanceMetric", "distanceMetricCache", "df", "y"] + return super()._tostring_excludes() + ["neighbor_provider_factory", "distance_metric", "distance_metric_cache", "df", "y"] # noinspection DuplicatedCode def _fit(self, x: pd.DataFrame, y: pd.DataFrame):