wip

microsoft · Oct 15, 2021 · c8d2177 · c8d2177
1 parent 466613a
commit c8d2177
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/python/synapse/ml/core/serialize/java_params_patch.py b/core/src/main/python/synapse/ml/core/serialize/java_params_patch.py
@@ -30,7 +30,7 @@ def __get_class(clazz):
         return m
 
     stage_name = java_stage.getClass().getName().replace("org.apache.spark", "pyspark")
-    stage_name = stage_name.replace("com.microsoft.azure.synapse.ml", "synapseml")
+    stage_name = stage_name.replace("com.microsoft.azure.synapse.ml", "synapse.ml")
     # Generate a default new instance from the stage_name class.
     py_type = __get_class(stage_name)
     if issubclass(py_type, JavaParams):