Skip to content

Commit d4c9dd4

Browse files
authored
Use Equals and = operator for DataViewType comparison (dotnet#5942)
1 parent eba938f commit d4c9dd4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/Microsoft.ML.AutoML/Utils/UserInputValidationUtil.cs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)