added correct validation check (dotnet#137)

srsaggam · web-flow · commit e6da4a46923f · 2019-02-14T14:31:53.000-08:00
diff --git a/src/Microsoft.ML.Auto/Utils/UserInputValidationUtil.cs b/src/Microsoft.ML.Auto/Utils/UserInputValidationUtil.cs
@@ -46,8 +46,8 @@ private static void ValidateTrainData(IDataView trainData)
             {
                 throw new ArgumentNullException(nameof(trainData), "Training data cannot be null");
             }
-
-            if (trainData.Schema.GetColumnOrNull(DefaultColumnNames.Features)?.Type.GetItemType() != NumberType.R4)
+            var type = trainData.Schema.GetColumnOrNull(DefaultColumnNames.Features)?.Type.GetItemType();
+            if (type != null && type != NumberType.R4)
             {
                 throw new ArgumentException($"{DefaultColumnNames.Features} column must be of data type Single", nameof(trainData));
             }

Original file line number	Diff line number	Diff line change
`@@ -46,8 +46,8 @@ private static void ValidateTrainData(IDataView trainData)`
`46`	`46`	`{`
`47`	`47`	`throw new ArgumentNullException(nameof(trainData), "Training data cannot be null");`
`48`	`48`	`}`
`49`		`-`
`50`		`- if (trainData.Schema.GetColumnOrNull(DefaultColumnNames.Features)?.Type.GetItemType() != NumberType.R4)`
	`49`	`+ var type = trainData.Schema.GetColumnOrNull(DefaultColumnNames.Features)?.Type.GetItemType();`
	`50`	`+ if (type != null && type != NumberType.R4)`
`51`	`51`	`{`
`52`	`52`	`throw new ArgumentException($"{DefaultColumnNames.Features} column must be of data type Single", nameof(trainData));`
`53`	`53`	`}`