Fix enum coercion to int

AmusementClub · Apr 4, 2024 · 6539ded · 6539ded
1 parent 0c9b5da
commit 6539ded
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/cython/vapoursynth.pyx b/src/cython/vapoursynth.pyx
@@ -792,7 +792,7 @@ cdef void dictToMap(dict ndict, VSMap *inm, VSCore *core, const VSAPI *funcs) ex
         ckey = key.encode('utf-8')
         val = ndict[key]
 
-        if isinstance(val, (str, bytes, bytearray, RawNode, RawFrame)):
+        if isinstance(val, (str, bytes, bytearray, RawNode, RawFrame, enum.Flag)):
             val = [val]
         else:
             try:
@@ -840,7 +840,7 @@ cdef void typedDictToMap(dict ndict, dict atypes, VSMap *inm, VSCore *core, cons
         if val is None:
             continue
 
-        if isinstance(val, (str, bytes, bytearray, enum.Flag, RawNode, RawFrame)) or not isinstance(val, Iterable):
+        if isinstance(val, (str, bytes, bytearray, enum.Flag, RawNode, RawFrame, enum.Flag)) or not isinstance(val, Iterable):
             val = [val]
 
         for v in val:
@@ -1028,7 +1028,7 @@ cdef class FrameProps(object):
         cdef bytes b = name.encode('utf-8')
         cdef const VSAPI *funcs = self.funcs
         val = value
-        if isinstance(val, (str, bytes, bytearray, RawNode, RawFrame)):
+        if isinstance(val, (str, bytes, bytearray, RawNode, RawFrame, enum.Flag)):
             val = [val]
         else:
             try: