quantizers 1.2.0__tar.gz → 1.2.1__tar.gz

{quantizers-1.2.0/src/quantizers.egg-info → quantizers-1.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantizers
-Version: 1.2.0
+Version: 1.2.1
 Author-email: Chang Sun <chsun@cern.ch>
 License: GNU Lesser General Public License v3 (LGPLv3)
 Project-URL: repository, https://github.com/calad0i/quantizers
@@ -15,7 +15,7 @@ License-File: LICENSE
 Provides-Extra: dependencies
 Requires-Dist: numpy; extra == "dependencies"
 Provides-Extra: grad
-Requires-Dist: keras>=3; extra == "grad"
+Requires-Dist: keras>=3.10; extra == "grad"
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"

{quantizers-1.2.0 → quantizers-1.2.1}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "quantizers"
-version = "1.2.0"
+version = "1.2.1"
 authors = [{ name = "Chang Sun", email = "chsun@cern.ch" }]
 description = ""
 readme = "README.md"
@@ -36,7 +36,7 @@ write_to = "src/quantizers/_version.py"
 [project.optional-dependencies]
 dependencies = ["numpy"]
 
-grad = ['keras>=3']
+grad = ['keras>=3.10']
 test = [
   'pytest',
   'pytest-cov',

{quantizers-1.2.0 → quantizers-1.2.1}/src/quantizers/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.2.0'
-__version_tuple__ = version_tuple = (1, 2, 0)
+__version__ = version = '1.2.1'
+__version_tuple__ = version_tuple = (1, 2, 1)

{quantizers-1.2.0 → quantizers-1.2.1}/src/quantizers/fixed_point/_fixed_point_ops.py

@@ -3,7 +3,7 @@ from functools import wraps
 from typing import Any, TypeVar
 
 from keras import ops
-from keras.api.random import SeedGenerator
+from keras.random import SeedGenerator
 from numpy.typing import ArrayLike
 
 round_mode_registry: dict[str, Callable[[Any], Any]] = {}

{quantizers-1.2.0 → quantizers-1.2.1}/src/quantizers/stateful_quantizers.py

@@ -87,11 +87,11 @@ class FixedQ:
     def __call__(self, x):
         use_numpy = not is_tensor(x)
         k, i, f = self.signed, self.integer - self.signed, self.width - self.integer
-        k, i, f = float(k), float(i), float(f)
+        # k, i, f = float(k), float(i), float(f)
         if use_numpy:
             q = get_fixed_quantizer_np(self.rounding, self.overflow)
         else:
-            from keras.api import ops
+            from keras import ops
 
             k, i, f = ops.convert_to_tensor(k), ops.convert_to_tensor(i), ops.convert_to_tensor(f)
             q = get_fixed_quantizer(self.rounding, self.overflow)
@@ -131,7 +131,7 @@ class MinifloatQ:
         if use_numpy:
             return float_quantize_np(x, s, e, e0)
         else:
-            from keras.api import ops
+            from keras import ops
 
             s, e, e0 = ops.convert_to_tensor(s), ops.convert_to_tensor(e), ops.convert_to_tensor(e0)
             return float_quantize(x, s, e, e0)

{quantizers-1.2.0 → quantizers-1.2.1/src/quantizers.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantizers
-Version: 1.2.0
+Version: 1.2.1
 Author-email: Chang Sun <chsun@cern.ch>
 License: GNU Lesser General Public License v3 (LGPLv3)
 Project-URL: repository, https://github.com/calad0i/quantizers
@@ -15,7 +15,7 @@ License-File: LICENSE
 Provides-Extra: dependencies
 Requires-Dist: numpy; extra == "dependencies"
 Provides-Extra: grad
-Requires-Dist: keras>=3; extra == "grad"
+Requires-Dist: keras>=3.10; extra == "grad"
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"

{quantizers-1.2.0 → quantizers-1.2.1}/src/quantizers.egg-info/requires.txt

@@ -3,7 +3,7 @@
 numpy
 
 [grad]
-keras>=3
+keras>=3.10
 
 [test]
 pytest

{quantizers-1.2.0 → quantizers-1.2.1}/test/test_stateless_quantizer.py

@@ -4,7 +4,7 @@ import jax
 import jax.numpy as jnp
 import numpy as np
 import pytest
-from keras.api.random import SeedGenerator
+from keras.random import SeedGenerator
 
 from quantizers import (
     binary_quantize,