Added predict back in.

KateBlueSky · KateBlueSky · commit 06944c172997 · 2025-03-21T16:20:52.000-07:00
diff --git a/configs/spmd/large_scale/kmeans_narrow_weak.json b/configs/spmd/large_scale/kmeans_narrow_weak.json
@@ -10,7 +10,7 @@
 		    "n_clusters": 10,
 		    "random_state": 42
                 },
-                "estimator_methods": { "training": "fit", "inference": "" },
+                "estimator_methods": { "training": "fit", "inference": "predict" },
                 "sklearnex_context": { "use_raw_input": true }
             }
         },
diff --git a/configs/spmd/large_scale/kmeans_wide_weak.json b/configs/spmd/large_scale/kmeans_wide_weak.json
@@ -10,7 +10,7 @@
 		    "n_clusters": 10,
 		    "random_state": 42
                 },
-                "estimator_methods": { "training": "fit", "inference": "" },
+                "estimator_methods": { "training": "fit", "inference": "predict" },
                 "sklearnex_context": { "use_raw_input": true }
             }
         },

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@`
`10`	`10`	`"n_clusters": 10,`
`11`	`11`	`"random_state": 42`
`12`	`12`	`},`
`13`		`- "estimator_methods": { "training": "fit", "inference": "" },`
	`13`	`+ "estimator_methods": { "training": "fit", "inference": "predict" },`
`14`	`14`	`"sklearnex_context": { "use_raw_input": true }`
`15`	`15`	`}`
`16`	`16`	`},`