Add __str__ and __format__ methods for expected behavior

NREL · Nov 4, 2023 · 6ec7206 · 6ec7206
1 parent cc81ad4
commit 6ec7206
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sup3r/utilities/__init__.py b/sup3r/utilities/__init__.py
@@ -47,6 +47,12 @@ class ModuleName(str, Enum):
  VISUAL_QA = 'visual-qa'
  REGRID = 'regrid'
 
+ def __str__(self):
+ return self.value
+
+ def __format__(self, format_spec):
+ return str.__format__(self.value, format_spec)
+
  @classmethod
  def all_names(cls):
  """All module names.