@@ -102,7 +102,7 @@ private static void ValidateTrainData(IDataView trainData, ColumnInformation col
102102 }
103103 }
104104
105- private static void ValidateColumnInformation ( IDataView trainData , ColumnInformation columnInformation , TaskKind task )
105+ private static void ValidateColumnInformation ( IDataView trainData , ColumnInformation columnInformation , TaskKind task )
106106 {
107107 ValidateColumnInformation ( columnInformation ) ;
108108 ValidateTrainDataColumn ( trainData , columnInformation . LabelColumnName , LabelColumnPurposeName , GetAllowedLabelTypes ( task ) ) ;
@@ -217,7 +217,7 @@ private static void ValidateValidationData(IDataView trainData, IDataView valida
217217 throw new ArgumentException ( $ "{ schemaMismatchError } Column '{ trainCol . Name } ' exists in train data, but not in validation data.", nameof ( validationData ) ) ;
218218 }
219219
220- if ( trainCol . Type != validCol . Value . Type )
220+ if ( trainCol . Type != validCol . Value . Type && ! trainCol . Type . Equals ( validCol . Value . Type ) )
221221 {
222222 throw new ArgumentException ( $ "{ schemaMismatchError } Column '{ trainCol . Name } ' is of type { trainCol . Type } in train data, and type " +
223223 $ "{ validCol . Value . Type } in validation data.", nameof ( validationData ) ) ;
@@ -260,7 +260,7 @@ private static void ValidateTrainDataColumn(IDataView trainData, string columnNa
260260 throw new ArgumentException ( exceptionMessage ) ;
261261 }
262262
263- if ( allowedTypes == null )
263+ if ( allowedTypes == null )
264264 {
265265 return ;
266266 }
0 commit comments