From a461a8db9d9f550ad0b468a548f485e3c713f537 Mon Sep 17 00:00:00 2001 From: Hagen Wierstorf Date: Thu, 23 Oct 2025 11:07:29 +0200 Subject: [PATCH 1/5] Add num_workers argument to load() --- audmodel/core/api.py | 4 +++- audmodel/core/backend.py | 4 ++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/audmodel/core/api.py b/audmodel/core/api.py index 607284b..5901833 100644 --- a/audmodel/core/api.py +++ b/audmodel/core/api.py @@ -278,6 +278,7 @@ def load( *, cache_root: str | None = None, timeout: float = 14400, # 4 h + num_workers: int = 1, verbose: bool = False, ) -> str: r"""Download a model by its unique ID. @@ -294,6 +295,7 @@ def load( If not set :meth:`audmodel.default_cache_root` is used timeout: maximum time in seconds before giving up acquiring a lock + num_workers: number of parallel jobs verbose: show debug messages Returns: @@ -313,7 +315,7 @@ def load( """ cache_root = audeer.safe_path(cache_root or default_cache_root()) short_id, version = split_uid(uid, cache_root) - return get_archive(short_id, version, cache_root, timeout, verbose) + return get_archive(short_id, version, cache_root, timeout, num_workers, verbose) def meta( diff --git a/audmodel/core/backend.py b/audmodel/core/backend.py index 0f04c08..e69d61d 100644 --- a/audmodel/core/backend.py +++ b/audmodel/core/backend.py @@ -60,6 +60,7 @@ def get_archive( version: str, cache_root: str, timeout: float, + num_workers: int, verbose: bool, ) -> str: r"""Return backend and local archive path. @@ -70,6 +71,7 @@ def get_archive( cache_root: path of cache root timeout: maximum time in seconds before giving up acquiring a lock + num_workers: number of parallel jobs verbose: if ``True`` show message or progress bar when downloading file @@ -106,6 +108,7 @@ def get_archive( src_path, dst_path, version, + num_workers=num_workers, verbose=verbose, ) @@ -114,6 +117,7 @@ def get_archive( dst_path, tmp_root, keep_archive=False, + num_workers=num_workers, verbose=verbose, ) From 5ea114e2f6883418052a75c73a5f6c8f2150873b Mon Sep 17 00:00:00 2001 From: Hagen Wierstorf Date: Thu, 23 Oct 2025 11:07:42 +0200 Subject: [PATCH 2/5] Depend on audbackend dev branch --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index a1f4f08..d8a91ba 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -31,7 +31,7 @@ classifiers = [ ] requires-python = '>=3.10' dependencies = [ - 'audbackend[all] >=2.2.2', + 'audbackend[all] @ git+https://github.com/audeering/audbackend.git@workers-download', 'filelock', 'oyaml', ] From 24c34c1986a741282775465d26d5cde70d74ac75 Mon Sep 17 00:00:00 2001 From: Hagen Wierstorf Date: Tue, 11 Nov 2025 12:47:02 +0100 Subject: [PATCH 3/5] Add dependency on audeer PR --- pyproject.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/pyproject.toml b/pyproject.toml index d8a91ba..982c6af 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,6 +32,7 @@ classifiers = [ requires-python = '>=3.10' dependencies = [ 'audbackend[all] @ git+https://github.com/audeering/audbackend.git@workers-download', + 'audeer @ git+https://https://github.com/audeering/audeer.git@archive-num-workers-2', 'filelock', 'oyaml', ] From b45a5dc1b7ff13895292bd1789647de6b1833420 Mon Sep 17 00:00:00 2001 From: Hagen Wierstorf Date: Tue, 11 Nov 2025 11:52:51 +0000 Subject: [PATCH 4/5] Fix URL --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 982c6af..78e8a8f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,7 +32,7 @@ classifiers = [ requires-python = '>=3.10' dependencies = [ 'audbackend[all] @ git+https://github.com/audeering/audbackend.git@workers-download', - 'audeer @ git+https://https://github.com/audeering/audeer.git@archive-num-workers-2', + 'audeer @ git+https://github.com/audeering/audeer.git@archive-num-workers-2', 'filelock', 'oyaml', ] From 3fef2a7f720abeffc9b82ad36e0f065e5b93841b Mon Sep 17 00:00:00 2001 From: Hagen Wierstorf Date: Wed, 7 Jan 2026 17:11:03 +0100 Subject: [PATCH 5/5] Depend on audbackend[all]>=2.3.0 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 78e8a8f..2ec6aa9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -31,7 +31,7 @@ classifiers = [ ] requires-python = '>=3.10' dependencies = [ - 'audbackend[all] @ git+https://github.com/audeering/audbackend.git@workers-download', + 'audbackend[all] >=2.3.0', 'audeer @ git+https://github.com/audeering/audeer.git@archive-num-workers-2', 'filelock', 'oyaml',