From 64a0eae7581db4b49e554948ae821a2438ff458b Mon Sep 17 00:00:00 2001
From: Artur Niederfahrenhorst <artur@anyscale.com>
Date: Mon, 27 Jun 2022 17:25:19 +0200
Subject: [PATCH] simplexfix (#26122)

---
 rllib/utils/spaces/simplex.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/rllib/utils/spaces/simplex.py b/rllib/utils/spaces/simplex.py
index 13e0442ff..4f7d97676 100644
--- a/rllib/utils/spaces/simplex.py
+++ b/rllib/utils/spaces/simplex.py
@@ -23,7 +23,6 @@ class Simplex(gym.Space):
 
     def __init__(self, shape, concentration=None, dtype=np.float32):
         assert type(shape) in [tuple, list]
-        assert len(shape) == 2, ""
 
         super().__init__(shape, dtype)
         self.dim = self.shape[-1]