File tree 2 files changed +2
-3
lines changed
src/Microsoft.ML.Core/Properties
test/Microsoft.ML.Functional.Tests
2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change 13
13
[ assembly: InternalsVisibleTo ( assemblyName : "Microsoft.ML.InferenceTesting" + PublicKey . TestValue ) ]
14
14
[ assembly: InternalsVisibleTo ( assemblyName : "Microsoft.ML.StaticPipelineTesting" + PublicKey . TestValue ) ]
15
15
[ assembly: InternalsVisibleTo ( assemblyName : "Microsoft.ML.OnnxTransformerTest" + PublicKey . TestValue ) ]
16
- [ assembly: InternalsVisibleTo ( assemblyName : "Microsoft.ML.Functional.Tests" + PublicKey . TestValue ) ]
17
16
18
17
[ assembly: InternalsVisibleTo ( assemblyName : "Microsoft.ML.EntryPoints" + PublicKey . Value ) ]
19
18
[ assembly: InternalsVisibleTo ( assemblyName : "Microsoft.ML.Maml" + PublicKey . Value ) ]
Original file line number Diff line number Diff line change @@ -106,10 +106,10 @@ public void SaveAndLoadModelWithLoader()
106
106
// Without deserializing the loader from the model we lose the slot names.
107
107
data = ml . Data . LoadFromEnumerable ( new [ ] { new InputData ( ) } ) ;
108
108
data = loadedModelWithoutLoader . Transform ( data ) ;
109
- Assert . Null ( data . Schema [ "Features" ] . Annotations . Schema . GetColumnOrNull ( AnnotationUtils . Kinds . SlotNames ) ) ;
109
+ Assert . True ( ! data . Schema [ "Features" ] . HasSlotNames ( ) ) ;
110
110
111
111
data = loadedModel . Load ( file ) ;
112
- Assert . True ( data . Schema [ "Features" ] . HasSlotNames ( data . Schema [ "Features" ] . Type . GetValueCount ( ) ) ) ;
112
+ Assert . True ( data . Schema [ "Features" ] . HasSlotNames ( ) ) ;
113
113
VBuffer < ReadOnlyMemory < char > > slotNames = default ;
114
114
data . Schema [ "Features" ] . GetSlotNames ( ref slotNames ) ;
115
115
var ageIndex = FindIndex ( slotNames . GetValues ( ) , "age" ) ;
You can’t perform that action at this time.
0 commit comments