Fix missing grid header include in Python bindings

Without the include the just-in-time compilation does not understand the grid in some circumstances. @andreas.dedner can you have a look please?