Skip to content
Snippets Groups Projects
Commit 7d5173e1 authored by Shinya Shiroshita's avatar Shinya Shiroshita
Browse files

random_test

parent 6f4e4c97
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 4 additions and 3 deletions
test.py
+ 4
− 3
View file @ 7d5173e1
......@@ -211,8 +211,8 @@ class MineRLSQILBaselineAgent(MineRLAgentBase):
#####################################################################
# IMPORTANT: SET THIS VARIABLE WITH THE AGENT CLASS YOU ARE USING #
######################################################################
AGENT_TO_TEST = MineRLSQILBaselineAgent # MineRLMatrixAgent # MineRLMatrixAgent, MineRLRandomAgent, YourAgentHere
# AGENT_TO_TEST = MineRLSQILBaselineAgent # MineRLMatrixAgent # MineRLMatrixAgent, MineRLRandomAgent, YourAgentHere
AGENT_TO_TEST = MineRLRandomAgent
####################
......@@ -250,7 +250,8 @@ def main():
envs = [wrapper(gym.make(MINERL_GYM_ENV)) for _ in range(EVALUATION_THREAD_COUNT)]
# envs = [gym.make(MINERL_GYM_ENV) for _ in range(EVALUATION_THREAD_COUNT)]
agent = AGENT_TO_TEST(envs[0])
# agent = AGENT_TO_TEST(envs[0])
agent = AGENT_TO_TEST()
assert isinstance(agent, MineRLAgentBase)
agent.load_agent()
......
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