o Ni@sddlZddlmZmZeejddgejdejdgejdejddgejdgejgd ejdejgd ejdd Zejgd ejdZGd ddeZdS)N)_ButcherTableauRKAdaptiveStepsizeODESolver?g?)dtypegp??)`?rr)g`rr)alphabetac_solc_error)rr c@seZdZdZeZeZdS) Fehlberg2N)__name__ __module__ __qualname__order_FEHLBERG2_TABLEAUtableau _FE_C_MIDmidrrO/home/ubuntu/.local/lib/python3.10/site-packages/torchdiffeq/_impl/fehlberg2.pyrsr) torch rk_commonrrtensorfloat64rrrrrrrs