From af04c946d431c022d6cf31a984b12116f8e9a822 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 11 Dec 2024 22:37:10 +0100 Subject: [PATCH] Refactor choices, imports and naming --- facefusion/choices.py | 2 +- tests/test_execution.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/facefusion/choices.py b/facefusion/choices.py index e0e76a66..6e91560a 100755 --- a/facefusion/choices.py +++ b/facefusion/choices.py @@ -37,7 +37,7 @@ execution_provider_set : ExecutionProviderSet =\ 'rocm': 'ROCMExecutionProvider', 'tensorrt': 'TensorrtExecutionProvider' } -execution_provider : List[ExecutionProvider] = list(execution_provider_set.keys()) +execution_providers : List[ExecutionProvider] = list(execution_provider_set.keys()) download_provider_set : DownloadProviderSet =\ { 'github': 'https://github.com/facefusion/facefusion-assets/releases/download/{base_name}/{file_name}', diff --git a/tests/test_execution.py b/tests/test_execution.py index aa836faf..1d1743c9 100644 --- a/tests/test_execution.py +++ b/tests/test_execution.py @@ -20,4 +20,4 @@ def test_multiple_execution_providers() -> None: 'CPUExecutionProvider' ] - assert create_inference_execution_providers('1', ['cpu', 'cuda']) == execution_providers + assert create_inference_execution_providers('1', [ 'cpu', 'cuda' ]) == execution_providers