CLI IFileInfo null exception fix (dotnet#254)

daholste · Dmitry-A · commit 09b2ebd15866 · 2019-08-22T14:40:15.000-07:00
diff --git a/src/mlnet/Commands/New/NewCommandHandler.cs b/src/mlnet/Commands/New/NewCommandHandler.cs
@@ -113,7 +113,7 @@ internal void GenerateProject(ColumnInferenceResults columnInference, Pipeline p
                 {
                     TrainDataset = new SystemFileInfo(settings.Dataset),
                     MlTask = taskKind,
-                    TestDataset = new SystemFileInfo(settings.TestDataset),
+                    TestDataset = settings.TestDataset == null ? null :  new SystemFileInfo(settings.TestDataset),
                     OutputName = settings.Name,
                     OutputBaseDir = settings.OutputPath.FullName,
                     LabelName = labelName,

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ internal void GenerateProject(ColumnInferenceResults columnInference, Pipeline p`
`113`	`113`	`{`
`114`	`114`	`TrainDataset = new SystemFileInfo(settings.Dataset),`
`115`	`115`	`MlTask = taskKind,`
`116`		`- TestDataset = new SystemFileInfo(settings.TestDataset),`
	`116`	`+ TestDataset = settings.TestDataset == null ? null : new SystemFileInfo(settings.TestDataset),`
`117`	`117`	`OutputName = settings.Name,`
`118`	`118`	`OutputBaseDir = settings.OutputPath.FullName,`
`119`	`119`	`LabelName = labelName,`