Merge pull request #21 from EdanToledo/chore/cleanup

chore: remove commented code
EdanToledo · Mar 9, 2024 · 6bb37a3 · 6bb37a3
2 parents d89e829 + fbc7930
commit 6bb37a3
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/stoix/evaluator.py b/stoix/evaluator.py
@@ -92,11 +92,6 @@ def _env_step(eval_state: EvalState) -> EvalState:
                 policy_key,
             )
 
-            # if config.arch.evaluation_greedy:
-            #     action = pi.mode()
-            # else:
-            #     action = pi.sample(seed=policy_key)
-
             # Step environment.
             env_state, timestep = env.step(env_state, action.squeeze())
 
@@ -201,11 +196,6 @@ def _env_step(eval_state: RNNEvalState) -> RNNEvalState:
             # Run the network.
             hstate, action = rec_act_fn(params, hstate, ac_in, policy_key)
 
-            # if config.arch.evaluation_greedy:
-            #     action = pi.mode()
-            # else:
-            #     action = pi.sample(seed=policy_key)
-
             # Step environment.
             env_state, timestep = env.step(env_state, action[-1].squeeze(0))