diff --git a/facefusion/codecs/vpx_encoder.py b/facefusion/codecs/vpx_encoder.py index 8f61c560..c095bbca 100644 --- a/facefusion/codecs/vpx_encoder.py +++ b/facefusion/codecs/vpx_encoder.py @@ -3,7 +3,7 @@ import struct from typing import Optional from facefusion.libraries import vpx as vpx_module -from facefusion.types import BitRate, Resolution, VideoCodec, VpxEncoder, VxpVideoCodec +from facefusion.types import BitRate, Resolution, VpxEncoder, VxpVideoCodec def create(video_codec : VxpVideoCodec, frame_resolution : Resolution, bitrate : BitRate, thread_count : int, cpu_count : int) -> Optional[VpxEncoder]: diff --git a/tests/test_api_stream_video.py b/tests/test_api_stream_video.py index f3e61bc8..f316870b 100644 --- a/tests/test_api_stream_video.py +++ b/tests/test_api_stream_video.py @@ -152,7 +152,7 @@ def test_encode_and_decode_video_frame(video_codec : VideoCodec) -> None: assert create_hash(decode_buffer) == 'ff3ecb43' if video_codec == 'vp9': - assert create_hash(decode_buffer) == 'ff3ecb43' + assert create_hash(decode_buffer) == 'a994fa02' assert decode_video_frame(video_codec, video_decoder, bytes()) is None diff --git a/tests/test_codec_vpx_encoder.py b/tests/test_codec_vpx_encoder.py index b70c4c07..17deacbd 100644 --- a/tests/test_codec_vpx_encoder.py +++ b/tests/test_codec_vpx_encoder.py @@ -2,8 +2,6 @@ from unittest.mock import patch import cv2 import pytest - -from facefusion.types import VxpVideoCodec from tests.assert_helper import get_test_example_file, get_test_examples_directory from facefusion import state_manager @@ -12,6 +10,7 @@ from facefusion.common_helper import is_linux, is_macos, is_windows from facefusion.download import conditional_download from facefusion.hash_helper import create_hash from facefusion.libraries import vpx as vpx_module +from facefusion.types import VxpVideoCodec from facefusion.vision import read_video_frame