update snapshot

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

Hide file tree

Showing 3 changed files with 218 additions and 185 deletions.
diff --git a/src/ffmpeg/common/tests/__snapshots__/test_serialize.ambr b/src/ffmpeg/common/tests/__snapshots__/test_serialize.ambr
@@ -112,10 +112,16 @@
                               ],
                               "name": "trim",
                               "input_typings": [
-                                "video"
+                                {
+                                  "__class__": "ffmpeg.common.schema.StreamType",
+                                  "value": "video"
+                                }
                               ],
                               "output_typings": [
-                                "video"
+                                {
+                                  "__class__": "ffmpeg.common.schema.StreamType",
+                                  "value": "video"
+                                }
                               ]
                             },
                             "index": 0
@@ -148,10 +154,16 @@
                               ],
                               "name": "trim",
                               "input_typings": [
-                                "video"
+                                {
+                                  "__class__": "ffmpeg.common.schema.StreamType",
+                                  "value": "video"
+                                }
                               ],
                               "output_typings": [
-                                "video"
+                                {
+                                  "__class__": "ffmpeg.common.schema.StreamType",
+                                  "value": "video"
+                                }
                               ]
                             },
                             "index": 0
@@ -196,33 +208,54 @@
                         ],
                         "name": "hflip",
                         "input_typings": [
-                          "video"
+                          {
+                            "__class__": "ffmpeg.common.schema.StreamType",
+                            "value": "video"
+                          }
                         ],
                         "output_typings": [
-                          "video"
+                          {
+                            "__class__": "ffmpeg.common.schema.StreamType",
+                            "value": "video"
+                          }
                         ]
                       },
                       "index": 0
                     }
                   ],
                   "name": "overlay",
                   "input_typings": [
-                    "video",
-                    "video"
+                    {
+                      "__class__": "ffmpeg.common.schema.StreamType",
+                      "value": "video"
+                    },
+                    {
+                      "__class__": "ffmpeg.common.schema.StreamType",
+                      "value": "video"
+                    }
                   ],
                   "output_typings": [
-                    "video"
+                    {
+                      "__class__": "ffmpeg.common.schema.StreamType",
+                      "value": "video"
+                    }
                   ]
                 },
                 "index": 0
               }
             ],
             "name": "drawbox",
             "input_typings": [
-              "video"
+              {
+                "__class__": "ffmpeg.common.schema.StreamType",
+                "value": "video"
+              }
             ],
             "output_typings": [
-              "video"
+              {
+                "__class__": "ffmpeg.common.schema.StreamType",
+                "value": "video"
+              }
             ]
           },
           "index": 0

diff --git a/src/ffmpeg/dag/factory.py b/src/ffmpeg/dag/factory.py
@@ -11,12 +11,12 @@ def filter_node_factory(filter: FFMpegFilterDef, *inputs: FilterableStream, **kw
     if isinstance(filter.typings_input, str):
         input_typings = tuple(eval(filter.typings_input, {"StreamType": StreamType, "re": re, **kwargs}))
     else:
-        input_typings = filter.typings_input
+        input_typings = tuple(StreamType.video if k == "video" else StreamType.audio for k in filter.typings_input)
 
     if isinstance(filter.typings_output, str):
         output_typings = tuple(eval(filter.typings_output, {"StreamType": StreamType, "re": re, **kwargs}))
     else:
-        output_typings = filter.typings_output
+        output_typings = tuple(StreamType.video if k == "video" else StreamType.audio for k in filter.typings_output)
 
     return FilterNode(
         name=filter.name,