From a78e81b9e150cd7f35c0801e677754c7b91a041c Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Thu, 21 Jul 2022 09:42:25 +0200 Subject: [PATCH] Allowing prob squashing --- metastable_baselines/distributions/distributions.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/metastable_baselines/distributions/distributions.py b/metastable_baselines/distributions/distributions.py index ae9adba..93fccac 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -148,9 +148,6 @@ class UniversalGaussianDistribution(SB3_Distribution): self.distribution = None self.gaussian_actions = None - if self.prob_squashing_type != ProbSquashingType.NONE: - raise Exception('ProbSquasing is not yet implmenented!') - if use_sde: raise Exception('SDE is not yet implemented')