Commit 3265daa3 authored by Eric Hambro's avatar Eric Hambro
Browse files

Fix typo.

parent d1ac97bd
......@@ -3,7 +3,7 @@ import numpy as np
from tqdm import trange
from collections.abc import Iterable
class BactchedEnv:
class BatchedEnv:
def __init__(self, env_make_fn, num_envs=32):
"""
Creates multiple copies of the environment with the same env_make_fn function
......
......@@ -8,7 +8,7 @@
from submission_agent import SubmissionConfig, LocalEvaluationConfig
from rollout import run_batched_rollout
from nethack_baselines.utils.batched_env import BactchedEnv
from nethack_baselines.utils.batched_env import BatchedEnv
# Ideally you shouldn't need to change anything below
......@@ -23,7 +23,7 @@ def evaluate():
Agent = SubmissionConfig.Submision_Agent
evaluation_env_fn = add_evaluation_wrappers_fn(submission_env_make_fn)
batched_env = BactchedEnv(env_make_fn=evaluation_env_fn,
batched_env = BatchedEnv(env_make_fn=evaluation_env_fn,
num_envs=num_envs)
num_envs = batched_env.num_envs
......
......@@ -9,7 +9,7 @@
from tqdm import tqdm
import numpy as np
from envs.batched_env import BactchedEnv
from envs.batched_env import BatchedEnv
from submission_config import SubmissionConfig
NUM_ASSESSMENTS = 512
......@@ -66,7 +66,7 @@ if __name__ == "__main__":
NUM_PARALLEL_ENVIRONMENTS = SubmissionConfig.NUM_PARALLEL_ENVIRONMENTS
Agent = SubmissionConfig.Submision_Agent
batched_env = BactchedEnv(
batched_env = BatchedEnv(
env_make_fn=submission_env_make_fn, num_envs=NUM_PARALLEL_ENVIRONMENTS
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment