Instead of loading the volume load as a fixed (x,y,z) vector, load it as a space-dependent Python function