From 4067021ecf9e01c2571e1cc1969e57fb53f4eae8 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Mon, 14 Oct 2024 08:04:34 +0100 Subject: [PATCH] amend --- torchrl/envs/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrl/envs/common.py b/torchrl/envs/common.py index 3af759c52a9..31ff0b905af 100644 --- a/torchrl/envs/common.py +++ b/torchrl/envs/common.py @@ -853,7 +853,7 @@ def full_action_spec(self) -> Composite: """ full_action_spec = self.input_spec.get("full_action_spec", None) if full_action_spec is None: - full_action_spec = CompositeSpec(shape=self.batch_size, device=self.device) + full_action_spec = Composite(shape=self.batch_size, device=self.device) self.input_spec.unlock_() self.input_spec["full_action_spec"] = full_action_spec self.input_spec.lock_()