fix import

sbi-dev · Aug 2, 2023 · 5e3bd38 · 5e3bd38
1 parent 5aef90b
commit 5e3bd38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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