Skip to content

Commit 1894796

Browse files
authored
Initial solution (#553)
1 parent f1de75d commit 1894796

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

lean/commands/data/download.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
from lean.models.data import Dataset, DataFile, DatasetDateOption, DatasetTextOption, DatasetTextOptionTransform,OptionResult, Product
2626
from lean.models.logger import Option
2727
from lean.models.cli import cli_data_downloaders
28+
from lean.constants import LIST_PENDING_DATASETS
2829

2930
_data_information: Optional[QCDataInformation] = None
3031
_presigned_terms="""
@@ -390,7 +391,7 @@ def _get_available_datasets(organization: QCFullOrganization) -> List[Dataset]:
390391

391392
available_datasets = []
392393
for cloud_dataset in cloud_datasets:
393-
if cloud_dataset.delivery == QCDatasetDelivery.CloudOnly or cloud_dataset.pending:
394+
if cloud_dataset.delivery == QCDatasetDelivery.CloudOnly or (not LIST_PENDING_DATASETS and cloud_dataset.pending):
394395
continue
395396

396397
datasource = data_information.datasources.get(str(cloud_dataset.id), None)

lean/constants.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,3 +116,6 @@
116116
# platforms
117117
MODULE_CLI_PLATFORM = "cli"
118118
MODULE_CLOUD_PLATFORM = "cloud"
119+
120+
# List pending datasets
121+
LIST_PENDING_DATASETS = environ.get("LIST_PENDING_DATASETS", "false").lower() == "true"

0 commit comments

Comments
 (0)