Skip to content

Add bucket and credentials parameter to download_demo #2747

@amontanez24

Description

@amontanez24

Problem Description

We will want to make it possible to download private demos if you have SDV Enterprise.

Expected behavior

  • To enable this we need to add the following parameters to download_demo, get_available_demos, get_source and get_readme:
  • credentials: A dict containing the following keys -
    • username: The username used for the DataCebo PyPI
    • license_key: The license key used for the DataCebo PyPI
  • s3_bucket_name: The name of the bucket the datasets are on.

I should be able to run

from sdv.datasets.demo import download_demo

data, metadata = download_demo(
  modality='single_table',
  dataset_name='ing_SEPA_data',
  s3_bucket_name='sdv-customer-datasets-private',
  credentials={
    'username': '[email protected]',
    'license_key': <MY_LICENSE_KEY>
  }
)

If someone provides a bucket other than sdv-datasets-public or if they provide credentials, we should raise an error saying that they need to have SDV Enterprise to support that feature.

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions