Fix _compute_raw call in KMeans

intel · Apr 19, 2024 · b33e801 · b33e801
1 parent 834b94b
commit b33e801
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onedal/cluster/kmeans.py b/onedal/cluster/kmeans.py
@@ -89,7 +89,7 @@ def _tolerance(self, rtol, X_table, policy, dtype=np.float32):
         # TODO: Support CSR in Basic Statistics
         dummy = to_table(None)
         bs = self._get_basic_statistics_backend("variance")
-        res = bs.compute_raw(X_table, dummy, policy, dtype)
+        res = bs._compute_raw(X_table, dummy, policy, dtype)
         mean_var = from_table(res["variance"]).mean()
         return mean_var * rtol