diff --git a/bnpm/torch_helpers.py b/bnpm/torch_helpers.py
index b4abbaa..870bac8 100644
--- a/bnpm/torch_helpers.py
+++ b/bnpm/torch_helpers.py
@@ -11,7 +11,6 @@
 from torch.utils.data import Dataset
 import numpy as np
 from tqdm.auto import tqdm
-import tensordict
 
 from . import indexing
 from . import misc