Skip to content

Commit f29b826

Browse files
[BugFix] Fixed dm_control action type casting (#463)
1 parent 3da4d6a commit f29b826

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchrl/trainers/helpers/envs.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,9 @@ def make_env_transforms(
152152
double_to_float_list += [
153153
"reward",
154154
]
155+
double_to_float_list += [
156+
"action",
157+
]
155158
double_to_float_inv_list += ["action"] # DMControl requires double-precision
156159
if not from_pixels:
157160
selected_keys = [

0 commit comments

Comments
 (0)