to poetry and gymnasium

envs/atari.py

@@ -1,4 +1,4 @@
-import gym
+import gymnasium as gym
 import numpy as np
 
 
@@ -38,7 +38,7 @@ class Atari:
             from PIL import Image
 
             self._image = Image
-        import gym.envs.atari
+        import gymnasium as gym.envs.atari
 
         if name == "james_bond":
             name = "jamesbond"

envs/crafter.py

@@ -1,4 +1,4 @@
-import gym
+import gymnasium as gym
 import numpy as np
 
 

envs/dmc.py

@@ -1,4 +1,4 @@
-import gym
+import gymnasium as gym
 import numpy as np
 
 

envs/dmlab.py

@@ -1,4 +1,4 @@
-import gym
+import gymnasium as gym
 import numpy as np
 import deepmind_lab
 

envs/memorymaze.py

@@ -1,4 +1,4 @@
-import gym
+import gymnasium as gym
 import numpy as np
 
 ###from tf dreamerv2 code

envs/minecraft.py

@@ -1,7 +1,7 @@
 import numpy as np
 from . import minecraft_base
 
-import gym
+import gymnasium as gym
 
 
 def make_env(task, *args, **kwargs):

envs/minecraft_base.py

@@ -2,7 +2,7 @@ import logging
 import threading
 
 import numpy as np
-import gym
+import gymnasium as gym
 
 
 class MinecraftBase(gym.Env):

envs/wrappers.py

@@ -1,5 +1,5 @@
 import datetime
-import gym
+import gymnasium as gym
 import numpy as np
 import uuid
 

pyproject.toml

@@ -0,0 +1,45 @@
+[tool.poetry]
+name = "dreamerv3-torch"
+version = "0.1.0"
+description = ""
+authors = ["wassname <github@wassname.org>"]
+license = "MIT"
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.9"
+torch = {version = "^2.1.0+cu121", source = "pytorch"}
+pandas = "^1.2.4"
+matplotlib = "^3.5.0"
+"ruamel.yaml" = "^0.17.4"
+moviepy = "^1.0.3"
+einops = "^0.3.0"
+protobuf = "^3.10.0"
+gymnasium = {version = "^v0.29.1", extras = ["other"]}
+# gym = "^0.19.0"
+# mujoco = "^2.3.5"
+# dm_control = "^1.0.9"
+scipy = "^1.8.0"
+# memory_maze = "^1.0.3"
+# atari-py = "^0.2.9"
+crafter = "^1.8.0"
+opencv-python = "^4.7.0.72"
+numpy = "^1.21.0"
+tensorboard = "^2.6.0"
+# minerl = "^0.4.4"
+jaxtyping = "^0.2.2"
+# typeguard = "^4.2.20"
+beartype = "^0.18.5"
+tqdm = "^4.66.4"
+loguru = "^0.7.2"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"
+
+[[tool.poetry.source]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cu121"
+priority = "explicit"
+
+

requirements.txt

@@ -1,22 +0,0 @@
-setuptools==60.0.0
-torch==2.0.0
-torchvision==0.15.1
-pandas==1.2.4
-matplotlib==3.5.0
-ruamel.yaml==0.17.4
-moviepy==1.0.3
-einops==0.3.0
-protobuf==3.20.0
-gym==0.19.0
-mujoco==2.3.5
-dm_control==1.0.9
-scipy==1.8.0
-memory_maze==1.0.3
-atari-py==0.2.9
-crafter==1.8.0
-opencv-python==4.7.0.72
-numpy==1.21.0
-tensorboard
-# minerl==0.4.4
-# This was needed for minerl
-# conda install -c conda-forge openjdk=8