NetKet 3.22.0.dev2__tar.gz → 3.22.0.dev3__tar.gz

{netket-3.22.0.dev2 → netket-3.22.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: NetKet
-Version: 3.22.0.dev2
+Version: 3.22.0.dev3
 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/#
@@ -54,7 +54,7 @@ Requires-Dist: myst-nb<1.5,>=1.0; extra == 'docs'
 Requires-Dist: myst-parser<5.1.0,>=2.0.0; extra == 'docs'
 Requires-Dist: nbconvert>=7.16.6; extra == 'docs'
 Requires-Dist: sphinx-autodoc-typehints<4.0,>=1.22; extra == 'docs'
-Requires-Dist: sphinx-book-theme~=1.1.0; extra == 'docs'
+Requires-Dist: sphinx-book-theme<1.3,>=1.1; extra == 'docs'
 Requires-Dist: sphinx-copybutton>=0.5.2; extra == 'docs'
 Requires-Dist: sphinx-design>=0.4.0; extra == 'docs'
 Requires-Dist: sphinx<9.2.0,>=8.2.3; extra == 'docs'

{netket-3.22.0.dev2 → netket-3.22.0.dev3}/netket/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '3.22.0.dev2'
-__version_tuple__ = version_tuple = (3, 22, 0, 'dev2')
+__version__ = version = '3.22.0.dev3'
+__version_tuple__ = version_tuple = (3, 22, 0, 'dev3')
 
 __commit_id__ = commit_id = None

{netket-3.22.0.dev2 → netket-3.22.0.dev3}/netket/experimental/dynamics/_rk/_tableau.py

@@ -108,24 +108,29 @@ bt_feuler = TableauRKExplicit(
 
 
 bt_midpoint = TableauRKExplicit(
-                order = (2,),
+                order = (2,1),
                 a = jnp.array([[0,   0],
                                [1/2, 0]], dtype=default_dtype),
-                b = jnp.array( [0,   1], dtype=default_dtype),
+                b = jnp.array([[0,   1],   # midpoint, order 2
+                               [1,   0]]),  # Euler, order 1
                 c = jnp.array( [0, 1/2], dtype=default_dtype),
                 name = "Midpoint"
                 )
 
 
 bt_heun = TableauRKExplicit(
-                order = (2,),
+                order = (2,1),
                 a = jnp.array([[0,   0],
                                [1,   0]], dtype=default_dtype),
-                b = jnp.array( [1/2, 1/2], dtype=default_dtype),
+                b = jnp.array([[1/2, 1/2],   # Heun, order 2
+                               [1,   0  ]]),  # Euler, order 1
                 c = jnp.array( [0, 1], dtype=default_dtype),
                 name = "Heun"
                 )
 
+# Adaptive step:
+# Heun Euler https://en.wikipedia.org/wiki/Runge–Kutta_methods
+bt_rk12  = bt_heun
 
 bt_rk4  = TableauRKExplicit(
                 order = (4,),
@@ -138,20 +143,6 @@ bt_rk4  = TableauRKExplicit(
                 name = "RK4"
                 )
 
-
-# Adaptive step:
-# Heun Euler https://en.wikipedia.org/wiki/Runge–Kutta_methods
-bt_rk12  = TableauRKExplicit(
-                order = (2,1),
-                a = jnp.array([[0,   0],
-                               [1,   0]], dtype=default_dtype),
-                b = jnp.array([[1/2, 1/2],
-                               [1,   0]], dtype=default_dtype),
-                c = jnp.array( [0, 1], dtype=default_dtype),
-                name = "RK12"
-                )
-
-
 # Bogacki–Shampine coefficients
 bt_rk23  = TableauRKExplicit(
                 order = (3,2),

{netket-3.22.0.dev2 → netket-3.22.0.dev3}/pyproject.toml

@@ -76,7 +76,7 @@ docs = [
     "sphinxcontrib-jsmath~=1.0.1",
     "myst-parser>=2.0.0,<5.1.0",
     "myst-nb>=1.0,<1.5",
-    "sphinx-book-theme~=1.1.0",
+    "sphinx-book-theme>=1.1,<1.3",
     "sphinx-copybutton>=0.5.2",
     "nbconvert>=7.16.6",
     "ipywidgets>=8.1.7",