bugfix

models/PDNet.py

@@ -8,7 +8,7 @@ from deepinv.unfolded import unfolded_builder
 from deepinv.utils.phantoms import RandomPhantomDataset, SheppLoganDataset
 from deepinv.optim.optim_iterators import CPIteration, fStep, gStep
 from deepinv.optim import Prior, DataFidelity
-from deepinv.utils.tensorlist import TensorList
+from deepinv.utils import TensorList
 
 from physics.multiscale import MultiScaleLinearPhysics
 from models.heads import Heads, Tails, InHead, OutTail, ConvChannels, SNRModule, EquivConvModule, EquivHeads