env v0.13

2023-06-14 20:22:17 +08:00
parent b9120a7440
commit 1cf0149c10
4 changed files with 10 additions and 8 deletions
--- a/envs/wrappers.py
+++ b/envs/wrappers.py
@@ -155,8 +155,8 @@ class OneHotAction:
        index = np.argmax(action).astype(int)
        reference = np.zeros_like(action)
        reference[index] = 1
-        if not np.allclose(reference, action):
-            raise ValueError(f"Invalid one-hot action:\n{action}")
+        # if not np.allclose(reference, action):
+        #     raise ValueError(f"Invalid one-hot action:\n{action}")
        return self._env.step(index)

    def reset(self):