From 55eec08e86654f4617562dcf22665e1b69ab6f88 Mon Sep 17 00:00:00 2001 From: Benjamin Blankenmeister Date: Tue, 25 Jun 2024 16:01:28 -0400 Subject: [PATCH 1/2] fix additional location --- v03_pipeline/lib/misc/lookup.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/v03_pipeline/lib/misc/lookup.py b/v03_pipeline/lib/misc/lookup.py index 1c3884638..344a22433 100644 --- a/v03_pipeline/lib/misc/lookup.py +++ b/v03_pipeline/lib/misc/lookup.py @@ -78,10 +78,12 @@ def remove_family_guids( if project_guid not in hl.eval(ht.globals.project_families): return ht project_i = ht.project_guids.index(project_guid) - family_indexes_to_keep = hl.array( - hl.enumerate(ht.globals.project_families[project_guid]) - .filter(lambda item: ~family_guids.contains(item[1])) - .map(lambda item: item[0]), + family_indexes_to_keep = hl.eval( + hl.array( + hl.enumerate(ht.globals.project_families[project_guid]) + .filter(lambda item: ~family_guids.contains(item[1])) + .map(lambda item: item[0]), + ), ) ht = ht.annotate( project_stats=( @@ -90,8 +92,12 @@ def remove_family_guids( hl.if_else( i != project_i, fs, - family_indexes_to_keep.map( - lambda j: ht.project_stats[i][j], + ( + hl.array(family_indexes_to_keep).map( + lambda j: ht.project_stats[i][j], + ) + if len(family_indexes_to_keep) > 0 + else hl.empty_array(ht.project_stats[i].dtype.element_type), ), ) ), From da21b55f2905e0dcce0bb05d8e5ccb6194bfaf6e Mon Sep 17 00:00:00 2001 From: Benjamin Blankenmeister Date: Tue, 25 Jun 2024 16:25:26 -0400 Subject: [PATCH 2/2] remove comma --- v03_pipeline/lib/misc/lookup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/v03_pipeline/lib/misc/lookup.py b/v03_pipeline/lib/misc/lookup.py index 344a22433..e8c32ae8c 100644 --- a/v03_pipeline/lib/misc/lookup.py +++ b/v03_pipeline/lib/misc/lookup.py @@ -97,7 +97,7 @@ def remove_family_guids( lambda j: ht.project_stats[i][j], ) if len(family_indexes_to_keep) > 0 - else hl.empty_array(ht.project_stats[i].dtype.element_type), + else hl.empty_array(ht.project_stats[i].dtype.element_type) ), ) ),