diff --git a/cogment_lab/specs/spaces_serialization.py b/cogment_lab/specs/spaces_serialization.py index d2547a1..1b8a566 100644 --- a/cogment_lab/specs/spaces_serialization.py +++ b/cogment_lab/specs/spaces_serialization.py @@ -112,7 +112,7 @@ def deserialize_space(pb_space: Space) -> gym.Space: if space_kind == "tuple": tuple_space_pb = pb_space.tuple spaces = [deserialize_space(sub_space) for sub_space in tuple_space_pb.spaces] - return gym.spaces.Tuple(spaces) + return gym.spaces.Tuple(spaces=spaces) if space_kind == "text": text_space_pb = pb_space.text return gym.spaces.Text(