From 2c14edd3b01cf34e9bf6b55953024e60897dafb6 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sat, 3 Sep 2022 23:41:06 +0200 Subject: [PATCH] Fix: Dependency --- metastable_baselines/distributions/distributions.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/metastable_baselines/distributions/distributions.py b/metastable_baselines/distributions/distributions.py index e17fb56..f5acfa0 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -7,8 +7,6 @@ from torch import nn from torch.distributions import Normal, Independent, MultivariateNormal from math import pi -from misc import givens - from stable_baselines3.common.preprocessing import get_action_dim from stable_baselines3.common.distributions import sum_independent_dims @@ -23,6 +21,7 @@ from stable_baselines3.common.distributions import DiagGaussianDistribution from ..misc.tensor_ops import fill_triangular from ..misc.tanhBijector import TanhBijector +from ..misc import givens class Strength(Enum):