Skip to content

Commit 6b32fdb

Browse files
committed
Unfriend Functional.Tests
1 parent b3a4d09 commit 6b32fdb

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/Microsoft.ML.Core/Properties/AssemblyInfo.cs

-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
[assembly: InternalsVisibleTo(assemblyName: "Microsoft.ML.InferenceTesting" + PublicKey.TestValue)]
1414
[assembly: InternalsVisibleTo(assemblyName: "Microsoft.ML.StaticPipelineTesting" + PublicKey.TestValue)]
1515
[assembly: InternalsVisibleTo(assemblyName: "Microsoft.ML.OnnxTransformerTest" + PublicKey.TestValue)]
16-
[assembly: InternalsVisibleTo(assemblyName: "Microsoft.ML.Functional.Tests" + PublicKey.TestValue)]
1716

1817
[assembly: InternalsVisibleTo(assemblyName: "Microsoft.ML.EntryPoints" + PublicKey.Value)]
1918
[assembly: InternalsVisibleTo(assemblyName: "Microsoft.ML.Maml" + PublicKey.Value)]

test/Microsoft.ML.Functional.Tests/ModelLoading.cs

+2-2
Original file line numberDiff line numberDiff line change
@@ -106,10 +106,10 @@ public void SaveAndLoadModelWithLoader()
106106
// Without deserializing the loader from the model we lose the slot names.
107107
data = ml.Data.LoadFromEnumerable(new[] { new InputData() });
108108
data = loadedModelWithoutLoader.Transform(data);
109-
Assert.Null(data.Schema["Features"].Annotations.Schema.GetColumnOrNull(AnnotationUtils.Kinds.SlotNames));
109+
Assert.True(!data.Schema["Features"].HasSlotNames());
110110

111111
data = loadedModel.Load(file);
112-
Assert.True(data.Schema["Features"].HasSlotNames(data.Schema["Features"].Type.GetValueCount()));
112+
Assert.True(data.Schema["Features"].HasSlotNames());
113113
VBuffer<ReadOnlyMemory<char>> slotNames = default;
114114
data.Schema["Features"].GetSlotNames(ref slotNames);
115115
var ageIndex = FindIndex(slotNames.GetValues(), "age");

0 commit comments

Comments
 (0)