Skip to content

Commit 85bee22

Browse files
author
Pete Luferenko
committed
Fixed tests
1 parent 2591a47 commit 85bee22

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/Microsoft.ML.Data/Transforms/Normalizer.cs

+4-1
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,9 @@
1313
[assembly: LoadableClass(typeof(NormalizerTransformer), null, typeof(SignatureLoadModel),
1414
"", NormalizerTransformer.LoaderSignature)]
1515

16+
[assembly: LoadableClass(typeof(IRowMapper), typeof(NormalizerTransformer), null, typeof(SignatureLoadRowMapper),
17+
"", NormalizerTransformer.LoaderSignature)]
18+
1619
namespace Microsoft.ML.Runtime.Data
1720
{
1821
public sealed class Normalizer : IEstimator<NormalizerTransformer>
@@ -375,7 +378,7 @@ public static NormalizerTransformer Create(IHostEnvironment env, ModelLoadContex
375378
return new NormalizerTransformer(env.Register(nameof(NormalizerTransformer)), ctx);
376379
}
377380

378-
// Factory method for SignatureRowMapper.
381+
// Factory method for SignatureLoadRowMapper.
379382
public static IRowMapper Create(IHostEnvironment env, ModelLoadContext ctx, ISchema inputSchema)
380383
=> Create(env, ctx).MakeRowMapper(inputSchema);
381384

0 commit comments

Comments
 (0)