diff --git a/src/mlnet/Templates/Console/ModelProject.cs b/src/mlnet/Templates/Console/ModelProject.cs index f20b483152..e16f29c915 100644 --- a/src/mlnet/Templates/Console/ModelProject.cs +++ b/src/mlnet/Templates/Console/ModelProject.cs @@ -27,18 +27,15 @@ public virtual string TransformText() { this.Write("\r\n\r\n \r\n netc" + "oreapp2.1\r\n \r\n \r\n \r\n"); - if(IncludeLightGBMPackage){ - this.Write(" \r" + - "\n"); + "rence Include=\"Microsoft.ML\" Version=\"1.0.0\" />\r\n"); + if (IncludeLightGBMPackage) { + this.Write(" \r\n"); } - if(IncludeMklComponentsPackage){ - this.Write(" \r\n"); + if (IncludeMklComponentsPackage){ + this.Write(" \r\n"); } - if(IncludeFastTreePackage){ - this.Write(" \r" + - "\n"); + if (IncludeFastTreePackage){ + this.Write(" \r\n"); } this.Write(" \r\n\r\n \r\n \r\n PreserveNewest\r\n \r\n netcoreapp2.1 - -<# if(IncludeLightGBMPackage){ #> - + +<# if (IncludeLightGBMPackage) { #> + <#}#> -<# if(IncludeMklComponentsPackage){ #> - +<# if (IncludeMklComponentsPackage){ #> + <#}#> -<# if(IncludeFastTreePackage){ #> - +<# if (IncludeFastTreePackage){ #> + <#}#> diff --git a/src/mlnet/Templates/Console/PredictProject.cs b/src/mlnet/Templates/Console/PredictProject.cs index 9aefb31775..e5a2c7d112 100644 --- a/src/mlnet/Templates/Console/PredictProject.cs +++ b/src/mlnet/Templates/Console/PredictProject.cs @@ -28,19 +28,16 @@ public virtual string TransformText() { this.Write("\r\n\r\n \r\n Exe\r\n netcoreapp2.1\r\n \r\n \r\n \r\n"); - if(IncludeLightGBMPackage){ - this.Write(" \r" + - "\n"); + ">\r\n \r\n \r\n"); + if (IncludeLightGBMPackage){ + this.Write(" \r\n"); } - if(IncludeMklComponentsPackage){ - this.Write(" \r\n"); + if (IncludeMklComponentsPackage){ + this.Write(" \r\n"); } - if(IncludeFastTreePackage){ - this.Write(" \r" + - "\n"); + if (IncludeFastTreePackage){ + this.Write(" \r\n"); } this.Write(" \r\n \r\n netcoreapp2.1 - -<# if(IncludeLightGBMPackage){ #> - + +<# if (IncludeLightGBMPackage){ #> + <#}#> -<# if(IncludeMklComponentsPackage){ #> - +<# if (IncludeMklComponentsPackage){ #> + <#}#> -<# if(IncludeFastTreePackage){ #> - +<# if (IncludeFastTreePackage){ #> + <#}#> diff --git a/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppProjectFileContentTest.approved.txt b/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppProjectFileContentTest.approved.txt index 789c3638c0..12908b1088 100644 --- a/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppProjectFileContentTest.approved.txt +++ b/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppProjectFileContentTest.approved.txt @@ -5,9 +5,9 @@ netcoreapp2.1 - - - + + + diff --git a/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ModelProjectFileContentTest.approved.txt b/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ModelProjectFileContentTest.approved.txt index 88af41da81..e97ae2fa8b 100644 --- a/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ModelProjectFileContentTest.approved.txt +++ b/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ModelProjectFileContentTest.approved.txt @@ -4,10 +4,10 @@ netcoreapp2.1 - - - - + + + +