BM3D: fixed format checking for "pre" and "flt"

HomeOfVapourSynthEvolution · Jan 26, 2016 · 6f93643 · 6f93643
1 parent b2337a1
commit 6f93643
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mvsfunc.py b/mvsfunc.py
@@ -878,15 +878,15 @@ def BM3D(input, sigma=None, radius1=None, radius2=None, profile1=None, profile2=
     if pre is not None:
         if not isinstance(pre, vs.VideoNode):
             raise TypeError(funcName + ': \"pre\" must be a clip!')
-        if pre.format != sFormat:
+        if pre.format.id != sFormat.id:
             raise ValueError(funcName + ': clip \"pre\" must be of the same format as the input clip!')
         if pre.width != input.width or pre.height != input.height:
             raise ValueError(funcName + ': clip \"pre\" must be of the same size as the input clip!')
 
     if ref is not None:
         if not isinstance(ref, vs.VideoNode):
             raise TypeError(funcName + ': \"ref\" must be a clip!')
-        if ref.format != sFormat:
+        if ref.format.id != sFormat.id:
             raise ValueError(funcName + ': clip \"ref\" must be of the same format as the input clip!')
         if ref.width != input.width or ref.height != input.height:
             raise ValueError(funcName + ': clip \"ref\" must be of the same size as the input clip!')