From d589a70750a1c96d90a9f5526c09f8872037a3f1 Mon Sep 17 00:00:00 2001
From: RichieHakim <RichHakim@gmail.com>
Date: Thu, 26 Sep 2024 21:40:27 -0400
Subject: [PATCH] remove tensordict

---
 bnpm/torch_helpers.py | 1 -
 1 file changed, 1 deletion(-)

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