Fix ruff type comparison complaint

2023-10-10 13:17:09 +02:00 · 2023-10-10 13:17:09 +02:00 · a8ea6808c3
commit a8ea6808c3
parent 1bb52a6a5c
1 changed files with 1 additions and 1 deletions
--- a/tianshou/highlevel/params/policy_wrapper.py
+++ b/tianshou/highlevel/params/policy_wrapper.py
@ -54,7 +54,7 @@ class PolicyWrapperFactoryIntrinsicCuriosity(
    ) -> ICMPolicy:
        feature_net = self.feature_net_factory.create_module(envs, device)
        action_dim = envs.get_action_shape()
-        if type(action_dim) != int:
+        if not isinstance(action_dim, int):
            raise ValueError(f"Environment action shape must be an integer, got {action_dim}")
        feature_dim = feature_net.output_dim
        icm_net = IntrinsicCuriosityModule(