Update spmd part

examples/sklearnex/basic_statistics_spmd.py

@@ -58,7 +58,7 @@ def generate_data(par, size, seed=777):
 gtr_std = np.std(weighted_data, axis=0)
 
 bss = BasicStatisticsSpmd(["mean", "standard_deviation"])
-res = bss.compute(dpt_data, dpt_weights)
+bss.fit(dpt_data, dpt_weights)
 
-print(f"Computed mean on rank {rank}:\n", res["mean"])
-print(f"Computed std on rank {rank}:\n", res["standard_deviation"])
+print(f"Computed mean on rank {rank}:\n", bss.mean)
+print(f"Computed std on rank {rank}:\n", bss.standard_deviation)

onedal/spmd/basic_statistics/basic_statistics.py

@@ -14,6 +14,8 @@
 # limitations under the License.
 # ==============================================================================
 
+import warnings
+
 from onedal.basic_statistics import BasicStatistics as BasicStatistics_Batch
 
 from ..._device_offload import support_usm_ndarray
@@ -23,4 +25,13 @@
 class BasicStatistics(BaseEstimatorSPMD, BasicStatistics_Batch):
     @support_usm_ndarray()
     def compute(self, data, weights=None, queue=None):
+        warnings.warn(
+            "Method `compute` was deprecated in version 2024.3 and will be "
+            "removed in 2024.5. Use `fit` instead."
+        )
         return super().compute(data, weights=weights, queue=queue)
+
+    @support_usm_ndarray()
+    def fit(self, data, sample_weight=None, queue=None):
+        super().fit(data, sample_weight=sample_weight, queue=queue)
+        return self