Support for Gymnasium Tuple Space

cogment_lab/generated/cog_settings.py

@@ -1,17 +1,3 @@
-# Copyright 2024 AI Redefined Inc. <[email protected]>
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 from types import SimpleNamespace
 
 import cogment as _cog
@@ -22,11 +8,11 @@
 
 
 _player_class = _cog.actor.ActorClass(
-    name="player",
-    config_type=data_pb.AgentConfig,
-    action_space=data_pb.PlayerAction,
-    observation_space=data_pb.Observation,
-)
+            name="player",
+            config_type=data_pb.AgentConfig,
+            action_space=data_pb.PlayerAction,
+            observation_space=data_pb.Observation,
+            )
 
 
 actor_classes = _cog.actor.ActorClassList(_player_class)

cogment_lab/protos/spaces.proto

@@ -44,6 +44,13 @@ message Dict {
   repeated SubSpace spaces = 1;
 }
 
+message Tuple {
+  message SubSpace {
+    Space space = 1;
+  }
+  repeated SubSpace spaces = 2;
+}
+
 message Text {
   int32 max_length = 1;
   int32 min_length = 2;
@@ -57,6 +64,7 @@ message Space {
     Dict dict = 3;
     MultiBinary multi_binary = 4;
     MultiDiscrete multi_discrete = 5;
-    Text text = 6;
+    Tuple tuple = 6;
+    Text text = 7;
   }
 }