Slightly faster filtering

B612-Asteroid-Institute · Aug 14, 2024 · fc3f963 · fc3f963
1 parent 1006676
commit fc3f963
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",