[BugFix] Fix gym info scalar infos (#1842)

Vincent Moens · web-flow · commit c2fae321abe5 · 2024-01-26T14:16:49.000Z
diff --git a/torchrl/envs/gym_like.py b/torchrl/envs/gym_like.py
@@ -84,7 +84,7 @@ def __init__(
             _info_spec = None
         elif spec is None:
             _info_spec = CompositeSpec(
-                {key: UnboundedContinuousTensorSpec() for key in keys}, shape=[]
+                {key: UnboundedContinuousTensorSpec(()) for key in keys}, shape=[]
             )
         elif not isinstance(spec, CompositeSpec):
             if self.keys is not None and len(spec) != len(self.keys):

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ def __init__(`
`84`	`84`	`_info_spec = None`
`85`	`85`	`elif spec is None:`
`86`	`86`	`_info_spec = CompositeSpec(`
`87`		`- {key: UnboundedContinuousTensorSpec() for key in keys}, shape=[]`
	`87`	`+ {key: UnboundedContinuousTensorSpec(()) for key in keys}, shape=[]`
`88`	`88`	`)`
`89`	`89`	`elif not isinstance(spec, CompositeSpec):`
`90`	`90`	`if self.keys is not None and len(spec) != len(self.keys):`