commit 58b216ed70c92ae58af4b18dc3a1c9e5b35e1e8e
author Phoenix Meadowlark <meadowlark@google.com> Sun Feb 21 13:21:08 2021 -0800
committer GitHub <noreply@github.com> Sun Feb 21 13:21:08 2021 -0800
tree 8ab004d63e045d99b50fda5d0596465595fa8aa8
parent d2cb99664cd282aaa316e2fe4a481048cdcd90aa

Allow '-' in OutputFormat enum parsing (#4850)

bindings/python/pyiree/compiler2/core.py

diff --git a/bindings/python/pyiree/compiler2/core.py b/bindings/python/pyiree/compiler2/core.py
index eea79fd..d93b7a4 100644
--- a/bindings/python/pyiree/compiler2/core.py
+++ b/bindings/python/pyiree/compiler2/core.py
@@ -53,7 +53,7 @@
     """
     if isinstance(spec, OutputFormat):
       return spec
-    spec = spec.upper()
+    spec = spec.upper().replace("-", "_")
     if spec not in OutputFormat.__members__:
       raise ValueError(f"For output_format= argument, expected one of: "
                        f"{', '.join(OutputFormat.__members__.keys())}")