From c82e8bd2bbfab66d5c1014dbf429614cec963e60 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20W=C4=85troba?= <markowanga@gmail.com>
Date: Sun, 15 Jan 2023 19:28:16 +0100
Subject: [PATCH] download_dataset command
---
new_datasets/add_asr_to_queue.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/new_datasets/add_asr_to_queue.py b/new_datasets/add_asr_to_queue.py
index 8b3da42..f9f5d93 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:
--
GitLab