From b0a3799f56290311661f2704c69adf4215f58c11 Mon Sep 17 00:00:00 2001
From: vmoens <vincentmoens@gmail.com>
Date: Tue, 3 Oct 2023 09:14:58 -0400
Subject: [PATCH] amend

---
 examples/sac/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/sac/utils.py b/examples/sac/utils.py
index 7058b2488d0..ebbee32057b 100644
--- 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(),
         ),
     )