diff --git a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs
index 73235797e5..a0d9577bfd 100644
--- a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs
+++ b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs
@@ -200,7 +200,8 @@ internal TransformerChain(IHostEnvironment env, ModelLoadContext ctx)
LastTransformer = null;
}
- public void SaveTo(IHostEnvironment env, Stream outputStream)
+ [BestFriend]
+ internal void SaveTo(IHostEnvironment env, Stream outputStream)
{
using (var ch = env.Start("Saving pipeline"))
{
diff --git a/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs b/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs
index fcc0905bad..0443240e1d 100644
--- a/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs
+++ b/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs
@@ -53,21 +53,5 @@ internal ExplainabilityTransforms(ModelOperationsCatalog owner)
_env = owner._env;
}
}
-
- ///
- /// Create a prediction engine for one-time prediction.
- ///
- /// The class that defines the input data.
- /// The class that defines the output data.
- /// The transformer to use for prediction.
- /// Additional settings of the input schema.
- /// Additional settings of the output schema.
- public PredictionEngine CreatePredictionEngine(ITransformer transformer,
- SchemaDefinition inputSchemaDefinition = null, SchemaDefinition outputSchemaDefinition = null)
- where TSrc : class
- where TDst : class, new()
- {
- return new PredictionEngine(_env, transformer, false, inputSchemaDefinition, outputSchemaDefinition);
- }
}
}