Fix numpy 2 compatibility

PiperOrigin-RevId: 725567326

CHANGELOG.md

@@ -8,6 +8,10 @@ Changelog follow the https://keepachangelog.com/ standard (at least the headers)
 
 ## [Unreleased]
 
+## [1.1.1] - 2025-02-11
+
+*   Restore numpy 1.26 compatibility
+
 ## [1.1.0] - 2025-02-07
 
 *   Add `kd.nn.WrapperModule` to make a inner-module transparent with
@@ -105,7 +109,8 @@ Changelog follow the https://keepachangelog.com/ standard (at least the headers)
 
 <!-- mdlint off(LINK_UNUSED_ID) -->
 
-[Unreleased]: https://github.com/google-research/kauldron/compare/v1.1.0...HEAD
+[Unreleased]: https://github.com/google-research/kauldron/compare/v1.1.1...HEAD
+[1.1.1]: https://github.com/google-research/kauldron/releases/tag/v1.1.0...v1.1.1
 [1.1.0]: https://github.com/google-research/kauldron/releases/tag/v1.0.0...v1.1.0
 [1.0.0]: https://github.com/google-research/kauldron/releases/tag/v0.1.0...v1.0.0
 [0.1.0]: https://github.com/google-research/kauldron/releases/tag/v0.1.0

kauldron/__init__.py

@@ -21,7 +21,7 @@
 
 # A new PyPI release will be pushed everytime `__version__` is increased
 # When changing this, also update the CHANGELOG.md
-__version__ = '1.1.0'
+__version__ = '1.1.1'
 
 
 def __getattr__(name: str):  # pylint: disable=invalid-name

pyproject.toml

@@ -30,7 +30,7 @@ dependencies = [
     "lark",
     "mediapy",
     "ml_collections",
-    "numpy>2",
+    "numpy",
     "opencv-python",  # speeds up pygrain image loading
     "optax",
     "orbax-checkpoint",