pre-commit

cogment · Apr 3, 2024 · 1f308be · 1f308be
1 parent 1646eb9
commit 1f308be
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 10 deletions.
diff --git a/cogment_lab/generated/cog_settings.py b/cogment_lab/generated/cog_settings.py
@@ -1,10 +1,11 @@
+from types import SimpleNamespace
 
 import cogment as _cog
-from types import SimpleNamespace
 
+import cogment_lab.generated.data_pb2 as data_pb
 import cogment_lab.generated.ndarray_pb2 as ndarray_pb
 import cogment_lab.generated.spaces_pb2 as spaces_pb
-import cogment_lab.generated.data_pb2 as data_pb
+
 
 _player_class = _cog.actor.ActorClass(
             name="player",

diff --git a/cogment_lab/generated/data_pb2.py b/cogment_lab/generated/data_pb2.py
diff --git a/cogment_lab/generated/ndarray_pb2.py b/cogment_lab/generated/ndarray_pb2.py
diff --git a/cogment_lab/generated/spaces_pb2.py b/cogment_lab/generated/spaces_pb2.py
diff --git a/cogment_lab/protos/spaces.proto b/cogment_lab/protos/spaces.proto
@@ -68,4 +68,3 @@ message Space {
     Text text = 7;
   }
 }
-
diff --git a/tests/test_spaces_serialization.py b/tests/test_spaces_serialization.py
@@ -12,14 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import gymnasium
 import numpy as np
 import pytest
 from gymnasium.spaces import Box, Dict, Discrete, MultiBinary, MultiDiscrete, Tuple
 
-
 from cogment_lab.specs.ndarray_serialization import SerializationFormat
-from cogment_lab.specs.spaces_serialization import deserialize_space, serialize_gym_space
+from cogment_lab.specs.spaces_serialization import serialize_gym_space
+
 
 # pylint: disable=no-member