NetKet 3.22.0.dev1__tar.gz → 3.22.0.dev2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: NetKet
-Version: 3.22.0.dev1
+Version: 3.22.0.dev2
 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.22.0.dev1 → netket-3.22.0.dev2}/netket/_src/driver/abstract_variational_driver.py

@@ -229,6 +229,21 @@ class AbstractDriver(struct.Pytree, mutable=True):
         self.reset_step()
         # self._step_count = 0
 
+    def _accept_step(self) -> bool:
+        """
+        Called after :meth:`compute_loss_and_update` to decide whether the
+        current step should be accepted.
+
+        Return ``False`` to reject the step and retry (``_step_attempt`` will
+        be incremented before the next attempt). Subclasses can override this
+        to implement driver-level acceptance criteria such as adaptive
+        step-size control: mutate driver state (e.g. ``self.dt``) as needed
+        before returning ``False``.
+
+        The default implementation always accepts.
+        """
+        return True
+
     def reset_step(self, hard: bool = False):
         """
         Resets the state of the driver at the beginning of a new step.
@@ -442,7 +457,8 @@ class AbstractDriver(struct.Pytree, mutable=True):
                         reject_step = callbacks.on_compute_update_end(
                             self.step_count, step_log_data, self
                         )
-                        if reject_step:
+                        driver_reject_step = not self._accept_step()
+                        if driver_reject_step or reject_step:
                             self._step_attempt += 1
                             continue
                         else:

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

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