drlab 0.2.8__tar.gz → 0.2.9__tar.gz

{drlab-0.2.8/src/drlab.egg-info → drlab-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drlab
-Version: 0.2.8
+Version: 0.2.9
 Summary: Deep Reinforcement Learning kit for research.
 Author: Tomas Osarte
 Keywords: reinforcement-learning,deep-learning,pytorch,gymnasium

{drlab-0.2.8 → drlab-0.2.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "drlab"
-version = "0.2.8"
+version = "0.2.9"
 description = "Deep Reinforcement Learning kit for research."
 readme = "README.md"
 requires-python = ">=3.11"

{drlab-0.2.8 → drlab-0.2.9}/src/drlab/__init__.py

@@ -35,7 +35,7 @@ from drlab.learners import (
 from drlab.replay import ReplayBuffer, TransitionBatch
 from drlab.runners import Runner
 
-__version__ = "0.2.8"
+__version__ = "0.2.9"
 
 __all__ = [
     "__version__",

{drlab-0.2.8 → drlab-0.2.9}/src/drlab/learners/off_policy/base.py

@@ -28,6 +28,10 @@ class OffPolicyLearner(ABC):
 
         self._validate_config()
 
+    def reset_optimizers(self) -> None:
+        """Reset optimizer state owned by the learner, if any."""
+        pass
+
     def _validate_config(self) -> None:
         if len(self.regularizers) != len(self.reg_lams):
             raise ValueError("regularizers and reg_lams must have same length.")

{drlab-0.2.8 → drlab-0.2.9/src/drlab.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drlab
-Version: 0.2.8
+Version: 0.2.9
 Summary: Deep Reinforcement Learning kit for research.
 Author: Tomas Osarte
 Keywords: reinforcement-learning,deep-learning,pytorch,gymnasium

{drlab-0.2.8 → drlab-0.2.9}/tests/test_off_policy_experiment.py

@@ -71,6 +71,11 @@ def one_transition() -> TransitionBatch:
 
 
 class OffPolicyExperimentTest(unittest.TestCase):
+    def test_base_learner_optimizer_reset_is_a_noop(self):
+        learner = CountingLearner()
+
+        self.assertIsNone(learner.reset_optimizers())
+
     def make_experiment(self, config: OffPolicyExperimentConfig) -> OffPolicyExperiment:
         env = OneStepEnv()
         learner = CountingLearner()

{drlab-0.2.8 → drlab-0.2.9}/tests/test_public_api.py

@@ -37,7 +37,7 @@ class PublicAPITest(unittest.TestCase):
             "WarmupController",
         }
 
-        self.assertEqual(drlab.__version__, "0.2.8")
+        self.assertEqual(drlab.__version__, "0.2.9")
         self.assertTrue(expected.issubset(set(drlab.__all__)))
         for name in expected:
             self.assertTrue(hasattr(drlab, name), name)