Skip to content

Commit a19a08e

Browse files
committed
Sync up with latest code.
1 parent f494e3b commit a19a08e

File tree

5 files changed

+8
-3
lines changed

5 files changed

+8
-3
lines changed

src/Microsoft.ML.Data/DataLoadSave/TransformWrapper.cs

+2-1
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,8 @@ private static VersionInfo GetVersionInfo()
7777
verWrittenCur: 0x00010001, // Initial
7878
verReadableCur: 0x00010001,
7979
verWeCanReadBack: 0x00010001,
80-
loaderSignature: LoaderSignature);
80+
loaderSignature: LoaderSignature,
81+
loaderAssemblyName: typeof(TransformWrapper).Assembly.FullName);
8182
}
8283

8384
// Factory for SignatureLoadModel.

src/Microsoft.ML.Data/Scorers/PredictionTransformer.cs

+2-1
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,8 @@ private static VersionInfo GetVersionInfo()
420420
verWrittenCur: 0x00010001, // Initial
421421
verReadableCur: 0x00010001,
422422
verWeCanReadBack: 0x00010001,
423-
loaderSignature: RankingPredictionTransformer.LoaderSignature);
423+
loaderSignature: RankingPredictionTransformer.LoaderSignature,
424+
loaderAssemblyName: typeof(RankingPredictionTransformer<>).Assembly.FullName);
424425
}
425426
}
426427

src/Microsoft.ML.StandardLearners/FactorizationMachine/FieldAwareFactorizationMachinePredictor.cs

+2-1
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,8 @@ private static VersionInfo GetVersionInfo()
354354
verWrittenCur: 0x00010001, // Initial
355355
verReadableCur: 0x00010001,
356356
verWeCanReadBack: 0x00010001,
357-
loaderSignature: LoaderSignature);
357+
loaderSignature: LoaderSignature,
358+
loaderAssemblyName: typeof(FieldAwareFactorizationMachinePredictionTransformer).Assembly.FullName);
358359
}
359360

360361
private static FieldAwareFactorizationMachinePredictionTransformer Create(IHostEnvironment env, ModelLoadContext ctx)

test/Microsoft.ML.Core.Tests/UnitTests/TestCSharpApi.cs

+1
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
using Microsoft.ML.Runtime.Data;
77
using Microsoft.ML.Runtime.EntryPoints;
88
using Microsoft.ML.TestFramework;
9+
using Microsoft.ML.Transforms;
910
using System;
1011
using System.Collections.Generic;
1112
using System.Linq;

test/Microsoft.ML.TestFramework/DataPipe/Parquet.cs

+1
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
// The .NET Foundation licenses this file to you under the MIT license.
33
// See the LICENSE file in the project root for more information.
44

5+
using Microsoft.ML.Runtime.Data;
56
using System;
67
using Xunit;
78

0 commit comments

Comments
 (0)