added minecraft environment

2023-07-02 11:29:48 +09:00
parent 8fa2274cfc
commit 036e9a8028
6 changed files with 543 additions and 5 deletions
--- a/dreamer.py
+++ b/dreamer.py
@@ -218,9 +218,12 @@ def make_env(config, logger, mode, train_eps, eval_eps):
        env = wrappers.OneHotAction(env)
    elif suite == "crafter":
        import envs.crafter as crafter
-
        env = crafter.Crafter(task, config.size)
        env = wrappers.OneHotAction(env)
+    elif suite == "minecraft":
+        import envs.minecraft as minecraft
+        env = minecraft.make_env(task, size=config.size, break_speed=config.break_speed)
+        env = wrappers.OneHotAction(env)
    else:
        raise NotImplementedError(suite)
    env = wrappers.TimeLimit(env, config.time_limit)