Added support for | union syntax

NAFTeam · Dec 19, 2021 · 831fe9f · 831fe9f
1 parent 68899a5
commit 831fe9f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/molter/command.py b/molter/command.py
@@ -2,6 +2,7 @@
 import inspect
 import typing
 from types import NoneType
+from types import UnionType
 
 import attr
 import dis_snek
@@ -72,7 +73,7 @@ def _get_params(func: typing.Callable):
 
         anno = param.annotation
 
-        if typing.get_origin(anno) == typing.Union:
+        if typing.get_origin(anno) in {typing.Union, UnionType}:
             for arg in typing.get_args(anno):
                 if arg != NoneType:
                     converter = _get_converter(arg)