Suggested best practice for switching between dolfin and dolfin_adjoint?

dolfin-adjoint
1

Hi all, I am looking for a good way to switch between using dolfin and dolfin_adjoint. I have a growing package and for several reasons (such as debugging, being able to use different builds of each between docker and anaconda/spyder, or not being able to call annotate=False when using dolfin, or avoiding star imports) I would like a way to specify which dolfin to import. For now, I am trying to use a boolean ie:

ADJOINT = False
from dolfin import Constant, det, tr, inner
if ADJOINT:
    from dolfin_adjoint import Constant

Would be curious if anyone could suggest better alternatives. Thanks in advance.