Merge pull request OpenMDAO#300 from xjjiang/cmd_line_verbosity

Command line verbosity option
xjjiang · Jun 5, 2024 · ed7e75d · ed7e75d
2 parents 7bd3f9f + 173c554
commit ed7e75d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/aviary/utils/fortran_to_aviary.py b/aviary/utils/fortran_to_aviary.py
@@ -612,7 +612,7 @@ def _setup_F2A_parser(parser):
         "--legacy_code",
         type=LegacyCode,
         help="Name of the legacy code the deck originated from",
-        choices=list(LegacyCode),
+        choices=set(LegacyCode),
         required=True
     )
     parser.add_argument(
@@ -629,8 +629,8 @@ def _setup_F2A_parser(parser):
     parser.add_argument(
         "-v",
         "--verbosity",
-        type=Verbosity,
-        choices=list(Verbosity),
+        type=int,
+        choices=Verbosity.set(),
         default=1,
         help="Set level of print statements",
     )

diff --git a/aviary/variable_info/enums.py b/aviary/variable_info/enums.py
@@ -195,3 +195,7 @@ class Verbosity(Enum):
 
     def __str__(self):
         return str(self.value)
+
+    @classmethod
+    def set(cls):
+        return {c.value for c in cls}