Skip to content

Commit 35b7dbb

Browse files
committed
file for speed testing
1 parent d0d648c commit 35b7dbb

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

speedtest.py

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
from nes_py import NESEnv
2+
import tqdm
3+
env = NESEnv('./nes_py/tests/games/super-mario-bros-1.nes')
4+
5+
done = True
6+
7+
try:
8+
for _ in tqdm.tqdm(range(5000)):
9+
if done:
10+
state = env.reset()
11+
done = False
12+
else:
13+
state, reward, done, info = env.step(env.action_space.sample())
14+
except KeyboardInterrupt:
15+
pass

0 commit comments

Comments
 (0)