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"