Skip to content

Commit a04dbf8

Browse files
authored
Internalize IDataSaver interface (#2329)
1 parent 6783a61 commit a04dbf8

File tree

6 files changed

+14
-9
lines changed

6 files changed

+14
-9
lines changed

src/Microsoft.ML.Data/Commands/SaveDataCommand.cs

+2-1
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,8 @@ private void RunCore(IChannel ch)
182182
}
183183
}
184184

185-
public static class DataSaverUtils
185+
[BestFriend]
186+
internal static class DataSaverUtils
186187
{
187188
public static void SaveDataView(IChannel ch, IDataSaver saver, IDataView view, IFileHandle file, bool keepHidden = false)
188189
{

src/Microsoft.ML.Data/Data/IDataLoader.cs

+4-2
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,11 @@ public interface IDataLoader : IDataView, ICanSaveModel
5555
{
5656
}
5757

58-
public delegate void SignatureDataSaver();
58+
[BestFriend]
59+
internal delegate void SignatureDataSaver();
5960

60-
public interface IDataSaver
61+
[BestFriend]
62+
internal interface IDataSaver
6163
{
6264
/// <summary>
6365
/// Check if the column can be saved.

src/Microsoft.ML.Data/DataLoadSave/Binary/BinarySaver.cs

+2-2
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
using System.Reflection;
1111
using System.Runtime.InteropServices;
1212
using System.Text;
13-
using System.Threading;
1413
using System.Threading.Tasks;
1514
using Microsoft.Data.DataView;
1615
using Microsoft.ML;
@@ -26,7 +25,8 @@ namespace Microsoft.ML.Data.IO
2625
{
2726
using Stopwatch = System.Diagnostics.Stopwatch;
2827

29-
public sealed class BinarySaver : IDataSaver
28+
[BestFriend]
29+
internal sealed class BinarySaver : IDataSaver
3030
{
3131
public sealed class Arguments
3232
{

src/Microsoft.ML.Data/DataLoadSave/Text/TextSaver.cs

+2-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919

2020
namespace Microsoft.ML.Data.IO
2121
{
22-
public sealed class TextSaver : IDataSaver
22+
[BestFriend]
23+
internal sealed class TextSaver : IDataSaver
2324
{
2425
// REVIEW: consider saving a command line in a separate file.
2526
public sealed class Arguments

src/Microsoft.ML.Data/DataLoadSave/Transpose/TransposeSaver.cs

+2-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@ namespace Microsoft.ML.Data.IO
2424
/// Saver for a format that can be loaded using the <see cref="TransposeLoader"/>.
2525
/// </summary>
2626
/// <seealso cref="TransposeLoader"/>
27-
public sealed class TransposeSaver : IDataSaver
27+
[BestFriend]
28+
internal sealed class TransposeSaver : IDataSaver
2829
{
2930
public sealed class Arguments
3031
{

src/Microsoft.ML.Data/Utilities/ComponentCreation.cs

+2-2
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ public static IDataLoader CreateLoader(this IHostEnvironment env, string setting
148148
/// <summary>
149149
/// Creates a data saver from the arguments object.
150150
/// </summary>
151-
public static IDataSaver CreateSaver<TArgs>(this IHostEnvironment env, TArgs arguments)
151+
internal static IDataSaver CreateSaver<TArgs>(this IHostEnvironment env, TArgs arguments)
152152
where TArgs : class, new()
153153
{
154154
Contracts.CheckValue(env, nameof(env));
@@ -158,7 +158,7 @@ public static IDataSaver CreateSaver<TArgs>(this IHostEnvironment env, TArgs arg
158158
/// <summary>
159159
/// Creates a data saver from the 'LoadName{settings}' string.
160160
/// </summary>
161-
public static IDataSaver CreateSaver(this IHostEnvironment env, string settings)
161+
internal static IDataSaver CreateSaver(this IHostEnvironment env, string settings)
162162
{
163163
Contracts.CheckValue(env, nameof(env));
164164
return CreateCore<IDataSaver>(env, typeof(SignatureDataSaver), settings);

0 commit comments

Comments
 (0)