NetKet 3.14.1__tar.gz → 3.14.2__tar.gz

{netket-3.14.1 → netket-3.14.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: NetKet
-Version: 3.14.1
+Version: 3.14.2
 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.14.1 → netket-3.14.2}/netket/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '3.14.1'
-__version_tuple__ = version_tuple = (3, 14, 1)
+__version__ = version = '3.14.2'
+__version_tuple__ = version_tuple = (3, 14, 2)

{netket-3.14.1 → netket-3.14.2}/netket/experimental/hilbert/spin_orbital_fermions.py

@@ -168,15 +168,16 @@ class SpinOrbitalFermions(HomogeneousHilbert):
                 if n_fermions is not None:
                     occupation_constraint = SumConstraint(n_fermions)
 
-        # Wrap the extra user provided cosntraint around our
-        # occupation constraint of the populations.
-        if constraint is not None:
-            constraint = ExtraConstraint(
-                base_constraint=occupation_constraint,
-                extra_constraint=constraint,
-            )
-        else:
-            constraint = occupation_constraint
+        if occupation_constraint is not None:
+            # Wrap the extra user provided cosntraint around our
+            # occupation constraint of the populations.
+            if constraint is not None:
+                constraint = ExtraConstraint(
+                    base_constraint=occupation_constraint,
+                    extra_constraint=constraint,
+                )
+            else:
+                constraint = occupation_constraint
 
         """Internal representation of this Hilbert space (Fock or TensorHilbert)."""
         # local states are the occupation numbers (0, 1)

{netket-3.14.1 → netket-3.14.2}/netket/hilbert/constraint/multiple.py

@@ -72,15 +72,11 @@ class ExtraConstraint(DiscreteHilbertConstraint):
         ):
             raise TypeError(
                 dedent(
-                    """The type parameters T1, T2 of ExtraConstraint[T1, T2] must be subtypes of
+                    """The arguments of ExtraConstraint must both be subtypes of
                     nk.hilbert.constraint.DiscreteHilbertConstraint, but one or more are not.
 
                     To verify if those are valid type parameters you can check it with
-                    issubclass(T1, nk.hilbert.constraint.DiscreteHilbertConstraint).
-
-                    For more information, look at the documentation of parametric classes in
-                    https://beartype.github.io/plum/parametric.html or open an issue on NetKet's
-                    GitHub discussion session.
+                    issubclass(arg, nk.hilbert.constraint.DiscreteHilbertConstraint).
                     """
                 )
             )
@@ -99,15 +95,11 @@ class ExtraConstraint(DiscreteHilbertConstraint):
         ):
             raise TypeError(
                 dedent(
-                    """The type parameters T1, T2 of ExtraConstraint[T1, T2] must be subtypes of
+                    """The arguments of ExtraConstraint must both be subtypes of
                     nk.hilbert.constraint.DiscreteHilbertConstraint, but one or more are not.
 
-                    To verify if those are valid type parameters you can check it with
-                    issubclass(T1, nk.hilbert.constraint.DiscreteHilbertConstraint).
-
-                    For more information, look at the documentation of parametric classes in
-                    https://beartype.github.io/plum/parametric.html or open an issue on NetKet's
-                    GitHub discussion session.
+                    To verify if those are valid arguments you can check it with
+                    isinstance(arg, nk.hilbert.constraint.DiscreteHilbertConstraint).
                     """
                 )
             )

{netket-3.14.1 → netket-3.14.2}/netket/vqs/full_summ/state.py

@@ -185,7 +185,7 @@ class FullSumState(VariationalState):
 
         key = nkjax.PRNGKey(seed)
 
-        dummy_input = jnp.zeros((1, self.hilbert.size), dtype=dtype)
+        dummy_input = self.hilbert.random_state(key, 1, dtype=dtype)
 
         variables = jit_evaluate(self._init_fun, {"params": key}, dummy_input)
         self.variables = variables

{netket-3.14.1 → netket-3.14.2}/netket/vqs/mc/mc_state/state.py

@@ -274,7 +274,7 @@ class MCState(VariationalState):
 
         key = nkjax.PRNGKey(seed)
 
-        dummy_input = jnp.zeros((1, self.hilbert.size), dtype=dtype)
+        dummy_input = self.hilbert.random_state(key, 1, dtype=dtype)
 
         variables = jit_evaluate(self._init_fun, {"params": key}, dummy_input)
         self.variables = variables

{netket-3.14.1 → netket-3.14.2}/pyproject.toml

@@ -101,6 +101,12 @@ source = "vcs"
 [tool.hatch.build.hooks.vcs]
 version-file = "netket/_version.py"
 
+[tool.uv]
+dev-dependencies = [
+  "netket[dev]",
+  "mpi4py",
+  "jax>=0.4.33"
+]
 
 # Development tools
 
@@ -145,7 +151,9 @@ filterwarnings = [
     "ignore:Defaulting to `holomorphic=False`,",
     "ignore::PendingDeprecationWarning: the matrix subclass",
     "ignore::netket.errors.HolomorphicUndeclaredWarning",
-    "ignore:netket.hilbert.CustomHilbert is deprecated"
+    "ignore:netket.hilbert.CustomHilbert is deprecated",
+    "ignore::netket.errors.UndeclaredSpinOderingWarning",
+    "ignore::netket.errors.UnoptimalSRtWarning",
 ]
 testpaths = [
     "test",