From 9fbdfecbfd167c937df0df5eb8e35c9c9afbacd2 Mon Sep 17 00:00:00 2001
From: Eric Hambro <eric.hambro@gmail.com>
Date: Sun, 6 Jun 2021 06:46:09 -0700
Subject: [PATCH] Fix up the ifname in env file.
---
envs/batched_env.py | 22 +++++++---------------
1 file changed, 7 insertions(+), 15 deletions(-)
diff --git a/envs/batched_env.py b/envs/batched_env.py
index 30a33b7..796e296 100644
--- a/envs/batched_env.py
+++ b/envs/batched_env.py
@@ -1,6 +1,6 @@
import aicrowd_gym
import numpy as np
-from tqdm import trange
+
from collections.abc import Iterable
class BatchedEnv:
@@ -52,23 +52,15 @@ class BatchedEnv:
if __name__ == '__main__':
- def nethack_make_fn():
- return aicrowd_gym.make('NetHackChallenge-v0',
- observation_keys=("glyphs",
- "chars",
- "colors",
- "specials",
- "blstats",
- "message",
- "tty_chars",
- "tty_colors",
- "tty_cursor",))
-
num_envs = 4
- batched_env = BactchedEnv(env_make_fn=nethack_make_fn, num_envs=num_envs)
+ batched_env = BatchedEnv(
+ env_make_fn=lambda:aicrowd_gym.make('NetHackChallenge-v0'),
+ num_envs=4
+ )
+
observations = batched_env.batch_reset()
num_actions = batched_env.envs[0].action_space.n
- for _ in trange(10000000000000):
+ for _ in range(50):
actions = np.random.randint(num_actions, size=num_envs)
observations, rewards, dones, infos = batched_env.batch_step(actions)
for done_idx in np.where(dones)[0]:
--
GitLab