amend

vmoens · vmoens · commit 314c069a8bc8 · 2024-02-06T11:10:40.000Z
diff --git a/torchrl/modules/models/multiagent.py b/torchrl/modules/models/multiagent.py
@@ -12,8 +12,7 @@
 
 from torchrl.data.utils import DEVICE_TYPING
 
-from .models import AbsLinear, HyperLinear, MLP
-from torchrl.modules.models import ConvNet, MLP
+from torchrl.modules.models import AbsLinear, ConvNet, HyperLinear, MLP
 
 
 class MultiAgentMLP(nn.Module):
