List valid vartypes at top of _base.py

smdogroup · Nov 9, 2023 · f0791d5 · f0791d5
1 parent 3e720d3
commit f0791d5
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/funtofem/model/_base.py b/funtofem/model/_base.py
@@ -28,6 +28,7 @@ class Base(object):
     """
     Base class for FUNtoFEM bodies and scenarios
     """
+    VAR_TYPES = ["structural", "aerodynamic", "rigid_motion", "shape", "control", "thermal"]
 
     def __init__(self, name, id=0, group=None):
         """
@@ -193,14 +194,7 @@ def verify_vartype(self, vartype):
             type of variable
         """
 
-        if not vartype in [
-            "structural",
-            "aerodynamic",
-            "rigid_motion",
-            "shape",
-            "controls",
-            "thermal",
-        ]:
+        if not vartype in self.VAR_TYPES:
             print(
                 f'Warning: vartype "{vartype}" is not a recognized variable type',
                 flush=True,