Skip to content
Snippets Groups Projects
Commit c82e8bd2 authored by Marcin Wątroba's avatar Marcin Wątroba
Browse files

download_dataset command

parent 7cde44f5
Branches
No related merge requests found
......@@ -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:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment