🐛 Fix value error string, value string in loads.py and element.py

Fusion-Power-Plant-Framework · Feb 28, 2024 · ff3bd82 · ff3bd82
1 parent 878ddf4
commit ff3bd82
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/bluemira/structural/element.py b/bluemira/structural/element.py
@@ -42,13 +42,13 @@ class LoadType(Enum):
     NODE_LOAD = auto()
 
     @classmethod
-    def _missing_(cls, value):
+    def _missing_(cls, value: str):
         try:
             return cls[value.replace(" ", "_").upper()]
         except KeyError:
             raise StructuralError(
-                f"Invalid load: {value}. Choose from:"
-                f"element_load, distributed_load, node_load"
+                f"{cls.__name__} has no load type {value}"
+                f"please select from {*cls._member_names_, }"
             ) from None
 
 

diff --git a/bluemira/structural/loads.py b/bluemira/structural/loads.py
@@ -25,12 +25,13 @@ class SubLoadType(Enum):
     ALL = auto()
 
     @classmethod
-    def _missing_(cls, value):
+    def _missing_(cls, value: str):
         try:
             return cls[value.upper()]
         except KeyError:
             raise StructuralError(
-                f"Unknown SubLoad type {value}. Choose from: force, moment or all"
+                f"{cls.__name__} has no load type {value}"
+                f"please select from {*cls._member_names_, }"
             ) from None