Use the -O flag for improved performance in error handling #31

mesa_frames/abstract/agents.py

@@ -998,10 +998,11 @@
 
     @abstractmethod
     def __getattr__(self, name: str) -> Any:
-        if name == "_agents":
-            raise RuntimeError(
-                "The _agents attribute is not set. You probably forgot to call super().__init__ in the __init__ method."
-            )
+        if __debug__:  # Only execute in non-optimized mode
+            if name == "_agents":
+                raise AttributeError(
+                    "The _agents attribute is not set. You probably forgot to call super().__init__ in the __init__ method."
+                )
 
     @overload
     def __getitem__(self, key: str | tuple[AgentMask, str]) -> Series | DataFrame: ...

mesa_frames/concrete/agents.py

@@ -438,10 +438,12 @@
         return super().__add__(other)
 
     def __getattr__(self, name: str) -> dict[AgentSetDF, Any]:
-        if name.startswith("_"):  # Avoids infinite recursion of private attributes
-            raise AttributeError(
-                f"'{self.__class__.__name__}' object has no attribute '{name}'"
-            )
+        # Avoids infinite recursion of private attributes
+        if __debug__:  # Only execute in non-optimized mode
+            if name.startswith("_"):
+                raise AttributeError(
+                    f"'{self.__class__.__name__}' object has no attribute '{name}'"
+                )
         return {agentset: getattr(agentset, name) for agentset in self._agentsets}
 
     @overload

mesa_frames/concrete/model.py

@@ -175,14 +175,17 @@
         try:
             return self._agents
         except AttributeError:
-            raise ValueError(
-                "You haven't called super().__init__() in your model. Make sure to call it in your __init__ method."
-            )
+            if __debug__:  # Only execute in non-optimized mode
+                raise RuntimeError(
+                    "You haven't called super().__init__() in your model. Make sure to call it in your __init__ method."
+                )
 
     @agents.setter
     def agents(self, agents: AgentsDF) -> None:
-        if not isinstance(agents, AgentsDF):
-            raise TypeError("agents must be an instance of AgentsDF")
+        if __debug__:  # Only execute in non-optimized mode
+            if not isinstance(agents, AgentsDF):
+                raise TypeError("agents must be an instance of AgentsDF")
+
         self._agents = agents
 
     @property