refactor(base_agents): change type checks to isinstance()

Met4FoF · May 14, 2024 · f9d3f3b · f9d3f3b
1 parent 3f25f43
commit f9d3f3b
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/agentMET4FOF/agents/base_agents.py b/agentMET4FOF/agents/base_agents.py
@@ -875,14 +875,16 @@ def get_all_attr(self):
             if key not in excludes and type(val).__name__ != "function"
         }
         filtered_attr = {
-            key: val
-            if (
-                type(val) == float
-                or type(val) == int
-                or type(val) == str
-                or key == "output_channels_info"
+            key: (
+                val
+                if (
+                    isinstance(val, float)
+                    or isinstance(val, int)
+                    or isinstance(val, str)
+                    or key == "output_channels_info"
+                )
+                else str(val)
             )
-            else str(val)
             for key, val in filtered_attr.items()
         }
         filtered_attr = {