NetKet 3.12.2__tar.gz → 3.12.3__tar.gz

{netket-3.12.2 → netket-3.12.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: NetKet
-Version: 3.12.2
+Version: 3.12.3
 Summary: Netket : Machine Learning toolbox for many-body quantum systems.
 Project-URL: homepage, https://www.netket.org
 Project-URL: documentation, https://netket.readthedocs.io/en/latest/#

{netket-3.12.2 → netket-3.12.3}/netket/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '3.12.2'
-__version_tuple__ = version_tuple = (3, 12, 2)
+__version__ = version = '3.12.3'
+__version_tuple__ = version_tuple = (3, 12, 3)

{netket-3.12.2 → netket-3.12.3}/netket/callbacks/earlystopping.py

@@ -78,7 +78,7 @@ class EarlyStopping(struct.Pytree, mutable=True):
         self.baseline = baseline
         self.monitor = monitor
 
-        self._best_val = np.infty
+        self._best_val = np.inf
         self._best_iter = 0
         self._best_patience_counter = 0
 

{netket-3.12.2 → netket-3.12.3}/netket/experimental/operator/_fermion_operator_2nd_base.py

@@ -488,7 +488,7 @@ class FermionOperator2ndBase(DiscreteOperator):
         del concrete
 
         # if operator is real dtype, then just return a copy
-        if not np.issubdtype(self.dtype, np.complex_):
+        if not np.issubdtype(self.dtype, np.complexfloating):
             return self.copy()
         else:
             new = type(self)(self.hilbert, dtype=self.dtype, cutoff=self._cutoff)

{netket-3.12.2 → netket-3.12.3}/netket/hilbert/random/base.py

@@ -23,7 +23,9 @@ import numpy as np
 from netket.utils.dispatch import dispatch
 from netket.jax.sharding import sharding_decorator
 
-Dim = Union[tuple[int], tuple[int, int], tuple[int, int, int]]
+Dim = Union[
+    tuple[int], tuple[int, int], tuple[int, int, int], tuple[int, int, int, int]
+]
 
 
 @dispatch
@@ -88,17 +90,6 @@ def random_state(hilb, key, size: int, *, dtype):  # noqa: F811
     )
 
 
-@dispatch
-def random_state(hilb, key, size: None, *, dtype):  # noqa: F811
-    return random_state(hilb, key, 1, dtype=dtype)[0]
-
-
-@dispatch
-def random_state(hilb, key, size: Dim, *, dtype):  # noqa: F811
-    n = int(np.prod(size))
-    return random_state(hilb, key, n, dtype=dtype).reshape(*size, -1)
-
-
 def flip_state(hilb, key, state, indices):
     r"""
     Given a state `σ` and an index `i`, randomly flips `σ[i]` so that

{netket-3.12.2 → netket-3.12.3}/netket/optimizer/qgt/qgt_jacobian_common.py

@@ -59,12 +59,14 @@ def sanitize_diag_shift(diag_shift, diag_scale, rescale_shift):
 
 
 def to_shift_offset(diag_shift, diag_scale):
-    if diag_scale == 0.0:
-        return diag_shift, None
+    if not isinstance(diag_scale, jax.Array):
+        if diag_scale == 0.0:
+            return diag_shift, None
+        else:
+            return diag_scale, diag_shift / diag_scale
     else:
         return diag_scale, diag_shift / diag_scale
 
-
 @partial(jax.jit, static_argnames="ndims")
 def rescale(centered_oks, offset, *, ndims: int = 1):
     """