Fix exlcluded torch.tensor

GilesStrong · Jul 7, 2021 · 4c3a2d9 · 4c3a2d9
1 parent e34f019
commit 4c3a2d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lumin/nn/data/batch_yielder.py b/lumin/nn/data/batch_yielder.py
@@ -5,7 +5,7 @@
 
 from ...utils.misc import to_device
 
-from torch.tensor import Tensor
+from torch import Tensor
 
 __all__ = ['BatchYielder']