From cb6f428d58e75afced674d6c65a2dafba026d839 Mon Sep 17 00:00:00 2001 From: kisnikser Date: Tue, 10 Dec 2024 12:57:53 +0300 Subject: [PATCH] fix torch import, change version --- src/relaxit/_version.py | 2 +- src/relaxit/distributions/LogisticNormalSoftmax.py | 1 + src/relaxit/distributions/kl.py | 1 + 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/relaxit/_version.py b/src/relaxit/_version.py index 545d07d..5da0b9a 100644 --- a/src/relaxit/_version.py +++ b/src/relaxit/_version.py @@ -1 +1 @@ -__version__ = "1.1.1" \ No newline at end of file +__version__ = "1.1.2" \ No newline at end of file diff --git a/src/relaxit/distributions/LogisticNormalSoftmax.py b/src/relaxit/distributions/LogisticNormalSoftmax.py index 377fa23..47f86f8 100644 --- 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 index 03b1e44..e0474ce 100644 --- 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