From 5adfeafc80330075a1166797770548c994a895de Mon Sep 17 00:00:00 2001 From: Steven Morad Date: Tue, 11 Jul 2023 14:26:03 +0100 Subject: [PATCH] i dont get import error, but try relative imports to fix ci fail --- tensordict/nn/ensemble.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensordict/nn/ensemble.py b/tensordict/nn/ensemble.py index 749ee8fef..bd6841948 100644 --- a/tensordict/nn/ensemble.py +++ b/tensordict/nn/ensemble.py @@ -1,8 +1,9 @@ import warnings import torch -from tensordict import TensorDict, TensorDictBase -from tensordict.nn import make_functional, TensorDictModuleBase +from tensordict import TensorDict +from .common import TensorDictBase, TensorDictModuleBase +from .functional_modules import make_functional from torch import nn