👽 Updates Oscillator environment version

rickstaa · Dec 3, 2020 · 9450234 · 9450234
1 parent 38a4dc4
commit 9450234
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 6 deletions.
diff --git a/.bumpversion.cfg b/.bumpversion.cfg
@@ -1,6 +1,7 @@
 [bumpversion]
 current_version = 0.0.3
 commit = True
+message = :bookmark: Bump version: {current_version} → {new_version}
 tag = True
 
-[bumpversion:file:setup.cfg]
+[bumpversion:file:simzoo.version.py]
diff --git a/simzoo/__init__.py b/simzoo/__init__.py
@@ -13,8 +13,9 @@
     namespace_prefix = "machine_learning_control.simzoo."
 
 envs = {
-    "name": ["Oscillator-v0", "Ex3_EKF-v0"],
+    "name": ["Oscillator-v1", "Ex3_EKF-v0"],
     "module": ["simzoo.envs.oscillator:Oscillator", "simzoo.envs.Ex3_EKF:Ex3_EKF"],
+    "max_step": [800, 800],
 }
 
 for idx, env in enumerate(envs["name"]):
@@ -24,5 +25,5 @@
         register(
             id=env,
             entry_point=namespace_prefix + envs["module"][idx],
-            max_episode_steps=400,
+            max_episode_steps=envs["max_step"][idx],
         )
diff --git a/simzoo/envs/oscillator b/simzoo/envs/oscillator
diff --git a/simzoo/envs/test.py b/simzoo/envs/test.py
@@ -0,0 +1,7 @@
+import gym
+
+# from register import Register
+from oscillator import Oscillator
+
+# register()
+gym.make("Oscillator-v0")
diff --git a/simzoo/version.py b/simzoo/version.py
@@ -1,4 +1,2 @@
 # coding: utf-8
-# file generated by setuptools_scm
-# don't change, don't track in version control
 version = '0.0.3'