From 12e03c04fb89edced7900e5de62f33f11d0e3860 Mon Sep 17 00:00:00 2001
From: tony <tqia574@aucklanduni.ac.nz>
Date: Thu, 9 Jan 2025 14:56:30 +1300
Subject: [PATCH] add yaos' for bounded exploration

---
 cares_reinforcement_learning/util/network_factory.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cares_reinforcement_learning/util/network_factory.py b/cares_reinforcement_learning/util/network_factory.py
index 6173845..b38474e 100644
--- 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.