Updated air hockey sac test

MushroomRL · Feb 5, 2025 · de6f703 · de6f703
1 parent a536821
commit de6f703
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/mujoco_air_hockey_sac.py b/examples/mujoco_air_hockey_sac.py
@@ -123,7 +123,7 @@ def experiment(alg, n_epochs, n_steps, n_steps_test):
 
     J = np.mean(dataset.discounted_return)
     R = np.mean(dataset.undiscounted_return)
-    E = agent.policy.entropy(dataset.state)
+    E = agent.policy.entropy(dataset.state).item()
 
     logger.epoch_info(0, J=J, R=R, entropy=E)
 
@@ -135,7 +135,7 @@ def experiment(alg, n_epochs, n_steps, n_steps_test):
 
         J = np.mean(dataset.discounted_return)
         R = np.mean(dataset.undiscounted_return)
-        E = agent.policy.entropy(dataset.state)
+        E = agent.policy.entropy(dataset.state).item()
 
         logger.epoch_info(n+1, J=J, R=R, entropy=E)