amend

pytorch · Oct 3, 2023 · b0a3799 · b0a3799
1 parent d0a6fab
commit b0a3799
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/sac/utils.py b/examples/sac/utils.py
@@ -39,7 +39,7 @@ def apply_env_transforms(env, max_episode_steps=1000):
         Compose(
             InitTracker(),
             StepCounter(max_episode_steps),
-            DoubleToFloat("observation"),
+            DoubleToFloat(),
             RewardSum(),
         ),
     )