From c0bcfb9e0e468a2c7ed8fd7fcb79ea690dda6aea Mon Sep 17 00:00:00 2001 From: Toni Kukurin Date: Wed, 3 Jun 2020 12:45:17 +0200 Subject: [PATCH 1/3] Minor fixes in block_pushing environment. max_episode_steps raises if info is None [1]. skimage import fix. [1] https://github.com/openai/gym/blob/master/gym/wrappers/time_limit.py#L19 --- envs/block_pushing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/envs/block_pushing.py b/envs/block_pushing.py index 94c5843..bf32b45 100644 --- a/envs/block_pushing.py +++ b/envs/block_pushing.py @@ -13,8 +13,8 @@ from mpl_toolkits.mplot3d import Axes3D from PIL import Image - import skimage +from skimage import draw def square(r0, c0, width, im_size): @@ -206,4 +206,4 @@ def step(self, action): state_obs = (self.get_state(), self.render()) - return state_obs, reward, done, None + return state_obs, reward, done, {} From 915bc9f57fb95e68a6ff6d78f6f8d13f3be1a4b4 Mon Sep 17 00:00:00 2001 From: Toni Kukurin Date: Wed, 3 Jun 2020 14:27:13 +0200 Subject: [PATCH 2/3] Use explicit skimage.draw import --- envs/block_pushing.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/envs/block_pushing.py b/envs/block_pushing.py index bf32b45..708957e 100644 --- a/envs/block_pushing.py +++ b/envs/block_pushing.py @@ -13,8 +13,7 @@ from mpl_toolkits.mplot3d import Axes3D from PIL import Image -import skimage -from skimage import draw +import skimage.draw def square(r0, c0, width, im_size): From 8f6bb1e1d45ac398eb914f1b15c60992bd157c79 Mon Sep 17 00:00:00 2001 From: Toni Kukurin Date: Wed, 3 Jun 2020 19:03:25 +0200 Subject: [PATCH 3/3] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d29d253..c062c9d 100644 --- a/README.md +++ b/README.md @@ -14,9 +14,9 @@ http://arxiv.org/abs/1911.12247 * Python 3.6 or 3.7 * PyTorch version 1.2 -* OpenAI Gym version: 0.12.0 `pip install gym==0.12.0` +* OpenAI Gym version: 0.17.2 `pip install gym==0.17.2` * OpenAI Atari_py version: 0.1.4: `pip install atari-py==0.1.4` -* Scikit-image version 0.15.0 `pip install scikit-image==0.15.0` +* Scikit-image version 0.17.2 `pip install scikit-image==0.17.2` * Matplotlib version 3.0.2 `pip install matplotlib==3.0.2` ## Generate datasets