From b0d7ddf327362b51fdf50f6b48ca0be0622c9b25 Mon Sep 17 00:00:00 2001 From: janfb Date: Wed, 2 Aug 2023 19:06:59 +0200 Subject: [PATCH] fix import --- tests/linearGaussian_snle_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/linearGaussian_snle_test.py b/tests/linearGaussian_snle_test.py index e8447fab9..242eb9220 100644 --- a/tests/linearGaussian_snle_test.py +++ b/tests/linearGaussian_snle_test.py @@ -4,6 +4,7 @@ from __future__ import annotations import pytest +import torch from torch import eye, ones, zeros from torch.distributions import HalfNormal, MultivariateNormal