diff --git a/new_datasets/add_asr_to_queue.py b/new_datasets/add_asr_to_queue.py index 8b3da423038b7aa6b1ba658944b05875b78eca4d..f9f5d93ee16163b484b3a7ac76eb636345da0aee 100644 --- a/new_datasets/add_asr_to_queue.py +++ b/new_datasets/add_asr_to_queue.py @@ -6,6 +6,8 @@ import pika from minio import Minio from pika.adapters.blocking_connection import BlockingChannel +from new_experiment.new_dependency_provider import get_experiment_repository + def get_all_datasets() -> List[str]: # return [it.object_name[:-1] for it in get_minio_client().list_objects('dataset-audio', '')] @@ -14,8 +16,7 @@ def get_all_datasets() -> List[str]: def get_dataset_items_id(dataset_name: str) -> List[str]: - return [it.object_name.split('/')[-1].split('.')[0] for it in - get_minio_client().list_objects('dataset-audio', f'{dataset_name}/')] + return list(get_experiment_repository(dataset_name).get_all_record_ids()) def get_minio_client() -> Minio: