local reboundop for now
@@ -4,10 +4,11 @@ __all__ = ["ReboundOrbit"]
import theano.tensor as tt
-from rebound_pymc3.integrate import IntegrateOp as ReboundOp
+# from rebound_pymc3.integrate import IntegrateOp as ReboundOp
from .keplerian import KeplerianOrbit
from .constants import day_per_yr_over_2pi, au_per_R_sun
+from ..theano_ops.rebound import ReboundOp
class ReboundOrbit(KeplerianOrbit):
