|
89 | 89 |
|
90 | 90 | import numpy as np |
91 | 91 | import soundfile as sf |
92 | | -from joblib import Parallel, delayed |
| 92 | +from joblib import Parallel, delayed, parallel_backend |
93 | 93 | from omegaconf import DictConfig, OmegaConf, open_dict |
94 | 94 | from tabulate import tabulate |
95 | 95 | from tqdm import tqdm |
@@ -292,7 +292,7 @@ def create_new_dataset( |
292 | 292 |
|
293 | 293 | manifest_folder, _ = os.path.split(manifest_path) |
294 | 294 |
|
295 | | - with parallel_backend(backend, n_jobs=num_workers): |
| 295 | + with parallel_backend(PARALLEL_BACKEND, n_jobs=num_workers): |
296 | 296 | new_entries_list = Parallel(verbose=config.num_shards)( |
297 | 297 | delayed(self._create_shard)(entries[start_idx:end_idx], target_dir, i, manifest_folder, only_manifests) |
298 | 298 | for i, (start_idx, end_idx) in enumerate(zip(start_indices, end_indices)) |
@@ -495,7 +495,7 @@ def create_concatenated_dataset( |
495 | 495 |
|
496 | 496 | manifest_folder, _ = os.path.split(base_manifest_path) |
497 | 497 |
|
498 | | - with parallel_backend(backend, n_jobs=num_workers): |
| 498 | + with parallel_backend(PARALLEL_BACKEND, n_jobs=num_workers): |
499 | 499 | new_entries_list = Parallel(verbose=config.num_shards)( |
500 | 500 | delayed(self._create_shard)( |
501 | 501 | entries[start_idx:end_idx], target_dir, shard_idx, manifest_folder, only_manifests |
|
0 commit comments