From f2d740abb6bb89dd84be076e4759c93f04817863 Mon Sep 17 00:00:00 2001 From: Nathan Thorpe Date: Tue, 22 Jul 2025 15:05:29 -0700 Subject: [PATCH] Check if instance of dataset with share in _format_share --- cirro/cli/interactive/download_args.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/cirro/cli/interactive/download_args.py b/cirro/cli/interactive/download_args.py index cdcfe322..fad0fa3b 100644 --- a/cirro/cli/interactive/download_args.py +++ b/cirro/cli/interactive/download_args.py @@ -12,11 +12,13 @@ from cirro.utils import format_date -def _format_share(dataset: DatasetWithShare) -> str: - return f'({dataset.share.name})' if dataset.share else '' +def _format_share(dataset: Dataset | DatasetWithShare) -> str: + if isinstance(dataset, DatasetWithShare) and dataset.share: + return f'({dataset.share.name})' + return '' -def ask_dataset(datasets: List[DatasetWithShare], input_value: str) -> str: +def ask_dataset(datasets: List[Dataset], input_value: str) -> str: if len(datasets) == 0: raise InputError("No datasets available") sorted_datasets = sorted(datasets, key=lambda d: d.created_at, reverse=True)