i dont get import error, but try relative imports to fix ci fail

pytorch · Jul 11, 2023 · 5adfeaf · 5adfeaf
1 parent b39d51b
commit 5adfeaf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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