test: update tests

UoA-CARES · Oct 31, 2023 · 7a3fe8a · 7a3fe8a
1 parent 322c09b
commit 7a3fe8a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cares_reinforcement_learning/util/NetworkFactory.py b/cares_reinforcement_learning/util/NetworkFactory.py
@@ -176,9 +176,14 @@ def create_network(self, observation_size, action_num, config: AlgorithmConfig):
         https://stackoverflow.com/questions/1796180/how-can-i-get-a-list-of-all-classes-within-current-module-in-python
         '''
 
+        agent = None
+
         for name, obj in inspect.getmembers(sys.modules[__name__]):
             if inspect.isfunction(obj):
                 if name == f"create_{algorithm}":
-                    return obj(observation_size, action_num, config)
+                    agent = obj(observation_size, action_num, config)
 
-        raise Exception(f"CARES RL NetworkFactory: {algorithm} is not implemented ")
+        if agent is None:
+            logging.warn(f"Unkown failed to return None: returned {agent}")
+
+        return agent