add yaos' for bounded exploration

UoA-CARES · Jan 9, 2025 · 12e03c0 · 12e03c0
1 parent eae247c
commit 12e03c0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cares_reinforcement_learning/util/network_factory.py b/cares_reinforcement_learning/util/network_factory.py
@@ -328,7 +328,9 @@ def create_STEVESAC_Bounded_Yao(
     return agent
 
 
-def create_DynaSAC_NS_IW(observation_size, action_num, config: acf.DynaSAC_NS_IWConfig):
+def create_DynaSAC_NS_IW(
+        observation_size, action_num, config: acf.DynaSAC_NS_IWConfig
+):
     """
     Create networks for model-based SAC agent. The Actor and Critic is same.
     An extra world model is added.