Why can't this simply be from `functions import time_derivative`? Then you would have circular dependencies and this could be moved to the import statements at the top of the file.