Can you elaborate more about what happens in the function? It's good you've identified the place it fails. Are you using ParallelRunner?