amend

pytorch · Oct 14, 2024 · 4067021 · 4067021
1 parent 04c327d
commit 4067021
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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_()