fix max_episode_steps condition

P6-Pool · Apr 15, 2024 · 83089b4 · 83089b4
1 parent 99a3ccc
commit 83089b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/fastfiz_env/envs/simple_fastfiz.py b/src/fastfiz_env/envs/simple_fastfiz.py
@@ -35,8 +35,8 @@ def __init__(
 
     def _max_episode_steps(self):
         if (
-            hasattr(SimpleFastFiz, "_time_limit_max_episode_steps")
-            and self.get_wrapper_attr("_time_limit_max_episode_steps") is not None
+            #hasattr(SimpleFastFiz, "_time_limit_max_episode_steps")
+            self.get_wrapper_attr("_time_limit_max_episode_steps") is not None
         ):
             self.max_episode_steps = self.get_wrapper_attr(
                 "_time_limit_max_episode_steps"

diff --git a/src/learn.py b/src/learn.py
@@ -6,7 +6,7 @@
 )
 from stable_baselines3 import PPO
 import fastfiz_env
-from fastfiz_env.reward_functions import DefaultReward, WinningReward
+from fastfiz_env.reward_functions import DefaultReward
 from fastfiz_env.reward_functions.common import *
 import os
 from fastfiz_env.wrappers import FastFizActionWrapper, ActionSpaces