From 252710489ec6bd31e5b094a53279999b28090d42 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 23 Jan 2024 09:22:58 +0100 Subject: [PATCH] Fixed error on clip_range None --- metastable_baselines2/trpl/trpl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/metastable_baselines2/trpl/trpl.py b/metastable_baselines2/trpl/trpl.py index ed819b1..043341a 100644 --- a/metastable_baselines2/trpl/trpl.py +++ b/metastable_baselines2/trpl/trpl.py @@ -186,7 +186,8 @@ class TRPL(BetterOnPolicyAlgorithm): super()._setup_model() # Initialize schedules for policy/value clipping - self.clip_range = get_schedule_fn(self.clip_range) + if self.clip_range is not None: + self.clip_range = get_schedule_fn(self.clip_range) if self.clip_range_vf is not None: if isinstance(self.clip_range_vf, (float, int)): assert self.clip_range_vf > 0, "`clip_range_vf` must be positive, " "pass `None` to deactivate vf clipping"