diff --git a/packages/data-designer/src/data_designer/cli/services/download_service.py b/packages/data-designer/src/data_designer/cli/services/download_service.py index be1594beb..749ed0872 100644 --- a/packages/data-designer/src/data_designer/cli/services/download_service.py +++ b/packages/data-designer/src/data_designer/cli/services/download_service.py @@ -53,6 +53,8 @@ def download_persona_dataset(self, locale: str) -> Path: "resource", "download-version", f"nvidia/nemotron-personas/{locale_obj.dataset_name}", + "--org", + "nvidia", "--dest", temp_dir, ] diff --git a/packages/data-designer/tests/cli/services/test_download_service.py b/packages/data-designer/tests/cli/services/test_download_service.py index 3abdaef75..fc59ac4f0 100644 --- a/packages/data-designer/tests/cli/services/test_download_service.py +++ b/packages/data-designer/tests/cli/services/test_download_service.py @@ -137,6 +137,8 @@ def test_download_persona_dataset_success( "resource", "download-version", "nvidia/nemotron-personas/nemotron-personas-dataset-en_us", + "--org", + "nvidia", "--dest", temp_dir_path, ]