Fix av library being a required import at startup

Move av import to be deferred in get_container_format() method to preserve
the optional nature of video dependencies. This allows ComfyUI to start
without av installed, only requiring it when video features are used.
This commit is contained in:
bymyself 2025-08-07 10:19:37 -07:00
parent 37d620a6b8
commit d9812c44b7

View File

@ -2,7 +2,6 @@ from __future__ import annotations
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import Optional, Union from typing import Optional, Union
import io import io
import av
from comfy_api.util import VideoContainer, VideoCodec, VideoComponents from comfy_api.util import VideoContainer, VideoCodec, VideoComponents
class VideoInput(ABC): class VideoInput(ABC):
@ -80,6 +79,7 @@ class VideoInput(ABC):
Container format as string Container format as string
""" """
# Default implementation - subclasses should override for better performance # Default implementation - subclasses should override for better performance
import av
source = self.get_stream_source() source = self.get_stream_source()
with av.open(source, mode="r") as container: with av.open(source, mode="r") as container:
return container.format.name return container.format.name