hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix ruff type comparison complaint

Browse Source
This commit is contained in:
Dominik Jain 2023-10-10 13:17:09 +02:00
parent 1bb52a6a5c
commit a8ea6808c3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tianshou/highlevel/params/policy_wrapper.py
View File

@ -54,7 +54,7 @@ class PolicyWrapperFactoryIntrinsicCuriosity(
) -> ICMPolicy: ) -> ICMPolicy:
feature_net = self.feature_net_factory.create_module(envs, device) feature_net = self.feature_net_factory.create_module(envs, device)
action_dim = envs.get_action_shape() 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}") raise ValueError(f"Environment action shape must be an integer, got {action_dim}")
feature_dim = feature_net.output_dim feature_dim = feature_net.output_dim
icm_net = IntrinsicCuriosityModule( icm_net = IntrinsicCuriosityModule(