Fix potential KeyError by using get() for target_accession_assembly in get_scoreset_metadata

bencap · bencap · commit 0ad233d54b8f · 2025-06-12T12:23:38.000-07:00
diff --git a/src/dcd_mapping/mavedb_data.py b/src/dcd_mapping/mavedb_data.py
@@ -220,7 +220,7 @@ def get_scoreset_metadata(
                     target_gene_name=gene["name"],
                     target_gene_category=gene["category"],
                     target_accession_id=target_accession_id,
-                    target_accession_assembly=target_gene_accession["assembly"],
+                    target_accession_assembly=target_gene_accession.get("assembly"),
                 )
         except (KeyError, ValidationError) as e:
             msg = f"Unable to extract metadata from API response for scoreset {scoreset_urn}: {e}"

Original file line number	Diff line number	Diff line change
`@@ -220,7 +220,7 @@ def get_scoreset_metadata(`
`220`	`220`	`target_gene_name=gene["name"],`
`221`	`221`	`target_gene_category=gene["category"],`
`222`	`222`	`target_accession_id=target_accession_id,`
`223`		`- target_accession_assembly=target_gene_accession["assembly"],`
	`223`	`+ target_accession_assembly=target_gene_accession.get("assembly"),`
`224`	`224`	`)`
`225`	`225`	`except (KeyError, ValidationError) as e:`
`226`	`226`	`msg = f"Unable to extract metadata from API response for scoreset {scoreset_urn}: {e}"`