def ode_fn(state):
     return geodesic_ode(times, state, metric_fn, metric_fn_kwargs)
 def ode_fn(state, t):
     print("inside ode_fn")
     return geodesic_ode(t, state, metric_fn, metric_fn_kwargs)
 def ode_fn(t, state):
     return geodesic_ode(t, state, metric_fn, metric_fn_args)
 def ode_fn_scipy(t, state):
     # print('inside ode_fn')
     return geodesic_ode(t, state, *ode_args)