From cfd42751470dd3f367979e509f489b9eeb8ebab1 Mon Sep 17 00:00:00 2001
From: Richard Stotz <richardstotz@google.com>
Date: Mon, 2 Dec 2024 07:59:44 -0800
Subject: [PATCH] [YDF] Prepare release of PYDF 0.9.0

PiperOrigin-RevId: 701968413
---
 yggdrasil_decision_forests/port/python/CHANGELOG.md      | 9 ++++++++-
 yggdrasil_decision_forests/port/python/config/setup.py   | 2 +-
 .../port/python/dev_requirements.txt                     | 2 +-
 .../port/python/tools/package_linux.sh                   | 4 ++--
 .../port/python/tools/release_windows.bat                | 2 +-
 yggdrasil_decision_forests/port/python/ydf/version.py    | 2 +-
 6 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/yggdrasil_decision_forests/port/python/CHANGELOG.md b/yggdrasil_decision_forests/port/python/CHANGELOG.md
index 39b0623b..1e09d9c7 100644
--- a/yggdrasil_decision_forests/port/python/CHANGELOG.md
+++ b/yggdrasil_decision_forests/port/python/CHANGELOG.md
@@ -1,6 +1,8 @@
 # Changelog
 
-## HEAD
+## 0.9.0 - 2024-12-02
+
+### Feature
 
 ### Breaking
 
@@ -23,11 +25,16 @@
 -   Add standalone prediction evaluation `ydf.evaluate_predictions()`.
 -   Add new hyperparameter `sparse_oblique_max_num_projections`.
 -   Add options "POWER_OF_TWO" and "INTEGER" for sparse oblique weights.
+-   Emit proper errors when using lists for multi-dimensional features.
 
 ### Fix
 
 -   Regression and Ranking CEPs scaling corrected.
 
+### Release music
+
+The John B. Sails. Traditional
+
 ## 0.8.0 - 2024-09-23
 
 ### Breaking
diff --git a/yggdrasil_decision_forests/port/python/config/setup.py b/yggdrasil_decision_forests/port/python/config/setup.py
index a6e5c3c3..e7bf9887 100644
--- a/yggdrasil_decision_forests/port/python/config/setup.py
+++ b/yggdrasil_decision_forests/port/python/config/setup.py
@@ -22,7 +22,7 @@
 from setuptools.command.install import install
 from setuptools.dist import Distribution
 
-_VERSION = "0.8.0"
+_VERSION = "0.9.0"
 
 with open("README.md", "r", encoding="utf-8") as fh:
   long_description = fh.read()
diff --git a/yggdrasil_decision_forests/port/python/dev_requirements.txt b/yggdrasil_decision_forests/port/python/dev_requirements.txt
index 9f802a59..d0d360bc 100644
--- a/yggdrasil_decision_forests/port/python/dev_requirements.txt
+++ b/yggdrasil_decision_forests/port/python/dev_requirements.txt
@@ -14,6 +14,6 @@ jaxlib; platform_machine != 'aarch64' and platform_system != 'Windows'
 optax; platform_machine != 'aarch64' and platform_system != 'Windows' and python_version >= '3.9'
 flatbuffers; platform_machine != 'aarch64' and platform_system != 'Windows' and python_version >= '3.12'
 tensorflow-datasets; platform_machine != 'aarch64' and platform_system != 'Windows' and python_version >= '3.9'
-grain; platform_machine != 'aarch64' and platform_system != 'Windows'
+grain; platform_machine != 'aarch64' and platform_system == 'Linux' and python_version >= '3.10'
 fastavro
 polars
diff --git a/yggdrasil_decision_forests/port/python/tools/package_linux.sh b/yggdrasil_decision_forests/port/python/tools/package_linux.sh
index 3ceb434d..ee08627b 100755
--- a/yggdrasil_decision_forests/port/python/tools/package_linux.sh
+++ b/yggdrasil_decision_forests/port/python/tools/package_linux.sh
@@ -108,13 +108,13 @@ function test_package() {
   local scaled_new_file_size=$(($new_file_size * 10))
   if [ "$scaled_new_file_size" -gt "$scaled_old_file_size" ]; then
     echo "New package is 20% larger than the previous one."
-    echo "This may indicates an issue with the wheel, aborting."
+    echo "This may indicate an issue with the wheel, aborting."
     exit 1
   fi
   scaled_old_file_size=$(($old_file_size * 8))
   if [ "$scaled_new_file_size" -lt "$scaled_old_file_size" ]; then
     echo "New package is 20% smaller than the previous one."
-    echo "This may indicates an issue with the wheel, aborting."
+    echo "This may indicate an issue with the wheel, aborting."
     exit 1
   fi
   rm -r previous_package
diff --git a/yggdrasil_decision_forests/port/python/tools/release_windows.bat b/yggdrasil_decision_forests/port/python/tools/release_windows.bat
index a391e85f..9561b85f 100644
--- a/yggdrasil_decision_forests/port/python/tools/release_windows.bat
+++ b/yggdrasil_decision_forests/port/python/tools/release_windows.bat
@@ -34,7 +34,7 @@
 cls
 setlocal
 
-set YDF_VERSION=0.8.0
+set YDF_VERSION=0.9.0
 set BAZEL=bazel.exe
 set BAZEL_SH=C:\msys64\usr\bin\bash.exe
 set BAZEL_FLAGS=--config=windows_cpp20 --config=windows_avx2
diff --git a/yggdrasil_decision_forests/port/python/ydf/version.py b/yggdrasil_decision_forests/port/python/ydf/version.py
index af929250..a49ac664 100644
--- a/yggdrasil_decision_forests/port/python/ydf/version.py
+++ b/yggdrasil_decision_forests/port/python/ydf/version.py
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-version = "0.8.0"
+version = "0.9.0"