From fc3f9631f4b9325e29207268b6f13d3324623dc9 Mon Sep 17 00:00:00 2001 From: Alec Koumjian Date: Wed, 14 Aug 2024 10:14:19 -0400 Subject: [PATCH] Slightly faster filtering --- src/adam_core/propagator/propagator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/adam_core/propagator/propagator.py b/src/adam_core/propagator/propagator.py index e31f304e..c0b1f671 100644 --- a/src/adam_core/propagator/propagator.py +++ b/src/adam_core/propagator/propagator.py @@ -584,8 +584,8 @@ def propagate_orbits( unique_origins = pc.unique(orbits.coordinates.origin.code) for origin_code in unique_origins: origin_orbits = orbits.select("coordinates.origin.code", origin_code) - result_origin_orbits = propagated.where( - pc.field("orbit_id").isin(origin_orbits.orbit_id) + result_origin_orbits = propagated.apply_mask( + pc.is_in(propagated.orbit_id, origin_orbits.orbit_id) ) partial_results = result_origin_orbits.set_column( "coordinates",