Remove print

zombie-einstein · Apr 22, 2023 · 2c925e0 · 2c925e0
1 parent 63d5bc8
commit 2c925e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax_ppo/mlp/training.py b/jax_ppo/mlp/training.py
@@ -131,7 +131,7 @@ def _step(carry, _):
     _, (state_series, reward_series, info_ts) = jax.lax.scan(
         _step, (key, agent, state, observation), None, length=n_steps
     )
-    print(reward_series.shape)
+
     return state_series, reward_series, info_ts