fix torch import, change version

intsystems · Dec 10, 2024 · cb6f428 · cb6f428
1 parent ab4a438
commit cb6f428
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/src/relaxit/_version.py b/src/relaxit/_version.py
@@ -1 +1 @@
-__version__ = "1.1.1"
+__version__ = "1.1.2"
diff --git a/src/relaxit/distributions/LogisticNormalSoftmax.py b/src/relaxit/distributions/LogisticNormalSoftmax.py
@@ -1,3 +1,4 @@
+import torch
 from pyro.distributions import constraints, Normal
 from pyro.distributions.torch import TransformedDistribution
 from pyro.distributions.transforms import SoftmaxTransform

diff --git a/src/relaxit/distributions/kl.py b/src/relaxit/distributions/kl.py
@@ -1,3 +1,4 @@
+import torch
 from torch.distributions import kl_divergence, register_kl, Normal
 from .InvertibleGaussian import InvertibleGaussian