KNearestNeighboursRegressionModel: Fix tostring excludes

opcode81 · Mar 2, 2024 · 9aa6474 · 9aa6474
1 parent 25ea617
commit 9aa6474
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):