From 086c695108c9c2676d7b014b58f960374bb44d6a Mon Sep 17 00:00:00 2001
From: rosebyte <jaroslav.ruzicka@hotmail.com>
Date: Thu, 17 Sep 2026 13:32:52 +0200
Subject: [PATCH] fix reference versions

---
 .../Program.cs                                        | 10 +++++-----
 test/NightlyBuildDependency.props                     | 11 +++++++++++
 test/TestFrameworkDependency.props                    |  1 +
 3 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/test/Microsoft.ML.NugetPackageVersionUpdater/Program.cs b/test/Microsoft.ML.NugetPackageVersionUpdater/Program.cs
index 0cf93f252c..97738ba725 100644
--- a/test/Microsoft.ML.NugetPackageVersionUpdater/Program.cs
+++ b/test/Microsoft.ML.NugetPackageVersionUpdater/Program.cs
@@ -50,7 +50,7 @@ private static IDictionary<string, string> GetLatestPackageVersions()
 
         private static void UpdatePackageVersion(string projectFiles, IDictionary<string, string> latestPackageVersions)
         {
-            string packageReferencePath = "/Project/ItemGroup/PackageReference";
+            string packageVersionPath = "/Project/ItemGroup/PackageVersion";
 
             var projectFilePaths = projectFiles.Split(";".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
 
@@ -59,16 +59,16 @@ private static void UpdatePackageVersion(string projectFiles, IDictionary<string
                 var csprojDoc = new XmlDocument();
                 csprojDoc.Load(projectFilePath);
 
-                var packageReferenceNodes = csprojDoc.DocumentElement.SelectNodes(packageReferencePath);
+                var packageVersionNodes = csprojDoc.DocumentElement.SelectNodes(packageVersionPath);
 
-                for (int i = 0; i < packageReferenceNodes.Count; i++)
+                for (int i = 0; i < packageVersionNodes.Count; i++)
                 {
-                    var packageName = packageReferenceNodes.Item(i).Attributes.GetNamedItem("Include").InnerText;
+                    var packageName = packageVersionNodes.Item(i).Attributes.GetNamedItem("Include").InnerText;
 
                     if (latestPackageVersions.ContainsKey(packageName))
                     {
                         var latestVersion = latestPackageVersions[packageName];
-                        packageReferenceNodes.Item(i).Attributes.GetNamedItem("Version").InnerText = latestVersion;
+                        packageVersionNodes.Item(i).Attributes.GetNamedItem("Version").InnerText = latestVersion;
                         Console.WriteLine($"Update packege {packageName} to version {latestVersion}.");
                     }
                     else
diff --git a/test/NightlyBuildDependency.props b/test/NightlyBuildDependency.props
index bb5a3d9618..9eb44ac034 100644
--- a/test/NightlyBuildDependency.props
+++ b/test/NightlyBuildDependency.props
@@ -2,6 +2,17 @@
   <ItemGroup>
     <!-- Versions below are just placeholder, will be updated during nighly build pipeline to latest package version
     -->
+    <PackageVersion Include="Microsoft.ML" Version="4.0.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.Ensemble" Version="0.22.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.EntryPoints" Version="0.22.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.FastTree" Version="4.0.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.LightGbm" Version="4.0.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.Mkl.Components" Version="4.0.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.OnnxConverter" Version="0.22.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.OnnxTransformer" Version="4.0.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.Parquet" Version="0.22.0-preview.24378.1" />
+    <PackageVersion Include="Microsoft.ML.Recommender" Version="0.22.0-preview.24378.1" />
+
     <PackageReference Include="Microsoft.ML" />
     <PackageReference Include="Microsoft.ML.Ensemble" />
     <PackageReference Include="Microsoft.ML.EntryPoints" />
diff --git a/test/TestFrameworkDependency.props b/test/TestFrameworkDependency.props
index 368a3dfb4a..7a4cdd2483 100644
--- a/test/TestFrameworkDependency.props
+++ b/test/TestFrameworkDependency.props
@@ -5,6 +5,7 @@
   <ItemGroup Condition="'$(ReferenceTypeForTestFramework)' == 'NuGet'">
     <!-- Versions below are just placeholder, will be updated during nighly build pipeline to latest package version
     -->
+    <PackageVersion Include="Microsoft.ML" Version="1.5.0-preview-28305-3" />
     <PackageReference Include="Microsoft.ML" />
   </ItemGroup>
 </Project>