update snapshot

livingbio · Mar 14, 2024 · 74efae5 · 74efae5
1 parent a1689ba
commit 74efae5
Show file tree

Hide file tree

Showing 9 changed files with 222 additions and 261 deletions.
diff --git a/src/ffmpeg/common/schema.py b/src/ffmpeg/common/schema.py
@@ -6,8 +6,14 @@
 
 
 class StreamType(str, Enum):
-    video = "video"
+    """
+    The type of a stream. (audio or video)
+    """
+
     audio = "audio"
+    """it is an audio stream"""
+    video = "video"
+    """it is a video stream"""
 
 
 class FFMpegFilterOptionType(str, Enum):

diff --git a/src/ffmpeg/common/tests/__snapshots__/test_serialize.ambr b/src/ffmpeg/common/tests/__snapshots__/test_serialize.ambr
@@ -112,16 +112,10 @@
                               ],
                               "name": "trim",
                               "input_typings": [
-                                {
-                                  "__class__": "ffmpeg.schema.StreamType",
-                                  "value": "video"
-                                }
+                                "video"
                               ],
                               "output_typings": [
-                                {
-                                  "__class__": "ffmpeg.schema.StreamType",
-                                  "value": "video"
-                                }
+                                "video"
                               ]
                             },
                             "index": 0
@@ -154,16 +148,10 @@
                               ],
                               "name": "trim",
                               "input_typings": [
-                                {
-                                  "__class__": "ffmpeg.schema.StreamType",
-                                  "value": "video"
-                                }
+                                "video"
                               ],
                               "output_typings": [
-                                {
-                                  "__class__": "ffmpeg.schema.StreamType",
-                                  "value": "video"
-                                }
+                                "video"
                               ]
                             },
                             "index": 0
@@ -172,17 +160,17 @@
                         "name": "concat",
                         "input_typings": [
                           {
-                            "__class__": "ffmpeg.schema.StreamType",
+                            "__class__": "ffmpeg.common.schema.StreamType",
                             "value": "video"
                           },
                           {
-                            "__class__": "ffmpeg.schema.StreamType",
+                            "__class__": "ffmpeg.common.schema.StreamType",
                             "value": "video"
                           }
                         ],
                         "output_typings": [
                           {
-                            "__class__": "ffmpeg.schema.StreamType",
+                            "__class__": "ffmpeg.common.schema.StreamType",
                             "value": "video"
                           }
                         ]
@@ -208,54 +196,33 @@
                         ],
                         "name": "hflip",
                         "input_typings": [
-                          {
-                            "__class__": "ffmpeg.schema.StreamType",
-                            "value": "video"
-                          }
+                          "video"
                         ],
                         "output_typings": [
-                          {
-                            "__class__": "ffmpeg.schema.StreamType",
-                            "value": "video"
-                          }
+                          "video"
                         ]
                       },
                       "index": 0
                     }
                   ],
                   "name": "overlay",
                   "input_typings": [
-                    {
-                      "__class__": "ffmpeg.schema.StreamType",
-                      "value": "video"
-                    },
-                    {
-                      "__class__": "ffmpeg.schema.StreamType",
-                      "value": "video"
-                    }
+                    "video",
+                    "video"
                   ],
                   "output_typings": [
-                    {
-                      "__class__": "ffmpeg.schema.StreamType",
-                      "value": "video"
-                    }
+                    "video"
                   ]
                 },
                 "index": 0
               }
             ],
             "name": "drawbox",
             "input_typings": [
-              {
-                "__class__": "ffmpeg.schema.StreamType",
-                "value": "video"
-              }
+              "video"
             ],
             "output_typings": [
-              {
-                "__class__": "ffmpeg.schema.StreamType",
-                "value": "video"
-              }
+              "video"
             ]
           },
           "index": 0

diff --git a/src/ffmpeg/dag/factory.py b/src/ffmpeg/dag/factory.py
@@ -9,19 +9,12 @@
 def filter_node_factory(filter: FFMpegFilterDef, *inputs: FilterableStream, **kwargs: Any) -> FilterNode:
 
     if isinstance(filter.typings_input, str):
-        input_typings = eval(
-            filter.typings_input,
-            {
-                "StreamType": StreamType,
-                "re": re,
-            }
-            | kwargs,
-        )
+        input_typings = tuple(eval(filter.typings_input, {"StreamType": StreamType, "re": re, **kwargs}))
     else:
         input_typings = filter.typings_input
 
     if isinstance(filter.typings_output, str):
-        output_typings = eval(filter.typings_output)
+        output_typings = tuple(eval(filter.typings_output, {"StreamType": StreamType, "re": re, **kwargs}))
     else:
         output_typings = filter.typings_output