diff --git a/metastable_baselines/distributions/distributions.py b/metastable_baselines/distributions/distributions.py index cfcf149..e17fb56 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -7,7 +7,7 @@ from torch import nn from torch.distributions import Normal, Independent, MultivariateNormal from math import pi -import givens +from misc import givens from stable_baselines3.common.preprocessing import get_action_dim