quantizers 1.2.0__tar.gz → 1.2.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantizers
-Version: 1.2.0
+Version: 1.2.2
 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.2}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "quantizers"
-version = "1.2.0"
+version = "1.2.2"
 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.2}/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.2'
+__version_tuple__ = version_tuple = (1, 2, 2)

{quantizers-1.2.0 → quantizers-1.2.2}/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]] = {}
@@ -124,7 +124,7 @@ def wrap_sm_fn(x, k, i, f, training=None, quant_fn: Callable = lambda x: x):
     eps = 2.0**-f
     high = 2.0**i - eps
     low = -(high + eps) * k
-    interval = 2 ** (i + k)
+    interval = 2.0 ** (i + k)
     c1 = ((x) / interval) % 2 >= 1  # type: ignore
     c1 = c1 & (ops.abs(x) > eps / 2)
     c2 = ((x + eps / 2) / (interval / 2)) % 2 >= 1  # type: ignore

{quantizers-1.2.0 → quantizers-1.2.2}/src/quantizers/fixed_point/fixed_point_ops_np.py

@@ -117,7 +117,7 @@ def wrap_sm_fn(x, k, i, f, training=None, quant_fn: Callable = lambda x: x):
     eps = 2.0**-f
     high = 2.0**i - eps
     low = -(high + eps) * k
-    interval = 2 ** (i + k)
+    interval = 2.0 ** (i + k)
     c1 = ((x) / interval) % 2 >= 1
     c1 = c1 & (np.abs(x) > eps / 2)
     c2 = ((x + eps / 2) / (interval / 2)) % 2 >= 1

{quantizers-1.2.0 → quantizers-1.2.2}/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.2/src/quantizers.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantizers
-Version: 1.2.0
+Version: 1.2.2
 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.2}/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.2}/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,