Auto-format code 🧹🌟🤖

UoA-CARES · Jan 9, 2025 · e253741 · e253741
1 parent 12e03c0
commit e253741
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/cares_reinforcement_learning/algorithm/mbrl/__init__.py b/cares_reinforcement_learning/algorithm/mbrl/__init__.py
@@ -7,6 +7,7 @@
 from .STEVESAC_Bounded import STEVESAC_Bounded
 from .DynaSAC_Bounded_Yao import DynaSAC_Bounded_Yao
 from .STEVESAC_Bounded_Yao import STEVESAC_Bounded_Yao
+
 # Immersive Weighting
 from .DynaSAC_NS_IW import DynaSAC_NS_IW
 from .DynaSAC_SUNRISE_NS import DynaSAC_SUNRISEReweight

diff --git a/cares_reinforcement_learning/util/configurations.py b/cares_reinforcement_learning/util/configurations.py
@@ -315,6 +315,7 @@ class STEVESAC_BoundedConfig_Yao(AlgorithmConfig):
     threshold: float = 0.1
     exploration_sample: int = 5
 
+
 class DynaSAC_Bounded_YaoConfig(AlgorithmConfig):
     algorithm: str = Field("DynaSAC_Bounded_Yao", Literal=True)
     type: str = Field("mbrl", Literal=True)

diff --git a/cares_reinforcement_learning/util/network_factory.py b/cares_reinforcement_learning/util/network_factory.py
@@ -328,9 +328,7 @@ 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.