Merge pull request #104 from sony/feature/20230714-fix-examples

Fix Deprecation error when using PendulumEnv
sony · Jul 21, 2023 · 81c1833 · 81c1833
2 parents 031f7fe + 6d79bcb
commit 81c1833
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 6 deletions.
diff --git a/examples/evaluate_trained_model.py b/examples/evaluate_trained_model.py
@@ -1,5 +1,5 @@
 # Copyright 2020,2021 Sony Corporation.
-# Copyright 2021 Sony Group Corporation.
+# Copyright 2021,2022,2023 Sony Group Corporation.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -21,7 +21,10 @@
 
 
 def build_env():
-    env = gym.make('Pendulum-v0')
+    try:
+        env = gym.make('Pendulum-v0')
+    except gym.error.DeprecatedEnv:
+        env = gym.make('Pendulum-v1')
     env = NumpyFloat32Env(env)
     env = ScreenRenderEnv(env)
     return env

diff --git a/examples/training_example.py b/examples/training_example.py
@@ -1,5 +1,5 @@
 # Copyright 2020,2021 Sony Corporation.
-# Copyright 2021 Sony Group Corporation.
+# Copyright 2021,2022,2023 Sony Group Corporation.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -22,7 +22,10 @@
 
 
 def build_env(seed=None):
-    env = gym.make('Pendulum-v0')
+    try:
+        env = gym.make('Pendulum-v0')
+    except gym.error.DeprecatedEnv:
+        env = gym.make('Pendulum-v1')
     env = NumpyFloat32Env(env)
     env = ScreenRenderEnv(env)
     env.seed(seed)

diff --git a/examples/writer_example.py b/examples/writer_example.py
@@ -1,5 +1,5 @@
 # Copyright 2020,2021 Sony Corporation.
-# Copyright 2021 Sony Group Corporation.
+# Copyright 2021,2022,2023 Sony Group Corporation.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -48,7 +48,10 @@ def _create_monitor_series(self, names):
 
 
 def build_env(seed=None):
-    env = gym.make('Pendulum-v0')
+    try:
+        env = gym.make('Pendulum-v0')
+    except gym.error.DeprecatedEnv:
+        env = gym.make('Pendulum-v1')
     env = NumpyFloat32Env(env)
     env.seed(seed)
     return env