Spaces:
Sleeping
Sleeping
Andrei Cozma
commited on
Commit
·
0a58f79
1
Parent(s):
ec9cd4e
Updates
Browse files- AgentBase.py +1 -2
AgentBase.py
CHANGED
|
@@ -88,8 +88,7 @@ class AgentBase:
|
|
| 88 |
def generate_episode(self, max_steps=500, render=False, **kwargs):
|
| 89 |
state, _ = self.env.reset()
|
| 90 |
# action = self.choose_action(state, **kwargs)
|
| 91 |
-
episode_hist = []
|
| 92 |
-
solved, done = False, False
|
| 93 |
rgb_array = self.env.render() if render else None
|
| 94 |
|
| 95 |
i = 0
|
|
|
|
| 88 |
def generate_episode(self, max_steps=500, render=False, **kwargs):
|
| 89 |
state, _ = self.env.reset()
|
| 90 |
# action = self.choose_action(state, **kwargs)
|
| 91 |
+
episode_hist, solved, done = [], False, False
|
|
|
|
| 92 |
rgb_array = self.env.render() if render else None
|
| 93 |
|
| 94 |
i = 0
|