o üN iã@s$ddlZddlZdd„Zdd„ZdS)éNc Cs¼t ¡Lt t |||¡t d¡¡}t| ¡ƒD]&}||d}||ƒ} t ||| ƒ¡} || k} t | ||¡}t | ||¡}q||d} Wdƒn1sSwY| || ƒfS)Ng@) ÚtorchÚno_gradÚceilÚlogÚmathÚrangeÚlongÚsignÚwhere) Ú interp_fnÚsign0Út0Út1Úevent_fnÚtolÚnitrsÚ_Út_midÚy_midÚsign_midÚ same_as_sign0Úevent_t©rúT/home/ubuntu/.local/lib/python3.10/site-packages/torchdiffeq/_impl/event_handling.pyÚ find_events " ô rcsJt ¡t ˆ||ƒ¡‰Wdƒn1swY‡‡fdd„}|S)zp We ensure all event functions are initially positive, so then we can combine them by taking a min. Ncsˆ||ƒ}t |ˆ¡S)N)rÚmin)ÚtÚyÚc©rÚ initial_signsrrÚcombined_event_fns z2combine_event_functions..combined_event_fn)rrr )rr Úy0r!rrrÚcombine_event_functionss ÿr#)rrrr#rrrrÚs