From 7be410dabc0d636a6581b9478ab2be896bc86ad4 Mon Sep 17 00:00:00 2001 From: Christian Bastian <80225746+cdb-boop@users.noreply.github.com> Date: Fri, 19 Jul 2024 03:29:32 -0400 Subject: [PATCH] Attempt to extract `bucket_info` from `ss_datasets`. --- __init__.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/__init__.py b/__init__.py index 313c988..53b1dbd 100644 --- a/__init__.py +++ b/__init__.py @@ -810,9 +810,16 @@ async def get_model_info(request): notes = f.read() if metadata is not None: - img_buckets = metadata.get("ss_bucket_info", "{}") + img_buckets = metadata.get("ss_bucket_info", None) + datasets = metadata.get("ss_datasets", None) + if type(img_buckets) is str: img_buckets = json.loads(img_buckets) + elif type(datasets) is str: + datasets = json.loads(datasets) + if isinstance(datasets, list): + datasets = datasets[0] + img_buckets = datasets.get("bucket_info", None) resolutions = {} if img_buckets is not None: buckets = img_buckets.get("buckets", {})