Merge pull request #184 from coreweave/eta/numpy

fix(serialization): Don't use obsolete `np.product` alias for `np.prod`

Author: wbrown

CHANGELOG.md

@@ -5,6 +5,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [Unreleased]
+
+### Fixed
+
+- Fixed compatibility with `numpy>=2.0.0`
+  - Calls to the removed `numpy.product` function now use `numpy.prod` instead
+
 ## [2.9.1] - 2024-11-27
 
 ### Fixed
@@ -409,6 +416,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   - `get_gpu_name`
   - `no_init_or_tensor`
 
+[Unreleased]: https://github.com/coreweave/tensorizer/compare/v2.9.1...HEAD
 [2.9.1]: https://github.com/coreweave/tensorizer/compare/v2.9.0...v2.9.1
 [2.9.0]: https://github.com/coreweave/tensorizer/compare/v2.8.1...v2.9.0
 [2.8.1]: https://github.com/coreweave/tensorizer/compare/v2.8.0...v2.8.1

tensorizer/_version.py

@@ -1 +1 @@
-__version__ = "2.9.1"
+__version__ = "2.9.2"

tensorizer/serialization.py

@@ -221,7 +221,7 @@ def deserialized_length(self):
         if self.data_length > 0:
             return self.data_length
         element_size: int = numpy.dtype(self.dtype).itemsize
-        num_elements: int = numpy.product(self.shape)
+        num_elements: int = numpy.prod(self.shape)
         return element_size * num_elements