Skip to content

Commit 468dee7

Browse files
committed
Speed up the event checking using numpy.
1 parent 181bf16 commit 468dee7

1 file changed

Lines changed: 3 additions & 9 deletions

File tree

kingmaker/wrapper.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -135,15 +135,9 @@ def __init__(
135135
def events_match(self, events: npt.NDArray[Any]):
136136
if self.events is None:
137137
return False
138-
try:
139-
events_match = True
140-
for key in self.events.keys():
141-
events_match &= all(self.events[key] == events[key])
142-
if not events_match:
143-
return False
144-
return events_match
145-
except ValueError:
146-
return np.array_equal(self.events, events)
138+
return np.array_equal(self.events["ra"], events["ra"]) & np.array_equal(
139+
self.events["dec"], events["dec"]
140+
)
147141

148142
def set_events(
149143
self,

0 commit comments

Comments
 (0)