From 4c3a2d957ddfa53847d7246c5c4ee94870c8ea31 Mon Sep 17 00:00:00 2001 From: GilesStrong Date: Wed, 7 Jul 2021 15:30:03 +0200 Subject: [PATCH] Fix exlcluded torch.tensor --- lumin/nn/data/batch_yielder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lumin/nn/data/batch_yielder.py b/lumin/nn/data/batch_yielder.py index 4e854a5..75a7fa8 100644 --- 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']